diff --git a/timesearch/commentaugment.py b/timesearch/commentaugment.py index 7191eb7..541141f 100644 --- a/timesearch/commentaugment.py +++ b/timesearch/commentaugment.py @@ -24,9 +24,8 @@ def commentaugment( common.r = common.bot.login(common.r) if specific_submission is not None: - if not specific_submission.startswith('t3_'): - specific_submission = 't3_' + specific_submission - specific_submission_obj = common.r.submission(specific_submission[3:]) + specific_submission = common.t3_prefix(specific_submission)[3:] + specific_submission_obj = common.r.submission(specific_submission) subreddit = specific_submission_obj.subreddit.display_name if subreddit: diff --git a/timesearch/common.py b/timesearch/common.py index e8f5ed3..dbb5aaa 100644 --- a/timesearch/common.py +++ b/timesearch/common.py @@ -119,3 +119,8 @@ def split_any(text, delimiters): for replacer in replacers: text = text.replace(replacer, splitter) return text.split(splitter) + +def t3_prefix(submission_id): + if not submission_id.startswith('t3_'): + submission_id = 't3_' + submission_id + return submission_id