Move t3_prefix to a helper function.

master
Ethan Dalool 2018-04-07 19:59:14 -07:00
parent c7dd46d6ec
commit 0267f221f0
2 changed files with 7 additions and 3 deletions

View File

@ -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:

View File

@ -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