def __init__(self, config):
self.refresh_token = config.refresh_token
self.card_limit = config.card_limit
self.post_limit = config.post_limit
self.debug = config.debug
self.stats = [
'skill',
'attack',
'energy',
'range',
'agility',
'hull',
'shield',
'points'
]
self.api = XWingAPI(config.api_base_url)
self.reddit = praw.Reddit(config.user_agent)
self.reddit.set_oauth_app_info(client_id=config.client_id,
client_secret=config.client_secret,
redirect_uri=config.redirect_uri)
self.sub = self.reddit.get_subreddit(config.subreddit)
class XWingTMGCardBot:
def __init__(self, config):
self.refresh_token = config.refresh_token
self.card_limit = config.card_limit
self.post_limit = config.post_limit
self.debug = config.debug
self.stats = [
'skill',
'attack',
'energy',
'range',
'agility',
'hull',
'shield',
'points'
]
self.api = XWingAPI(config.api_base_url)
self.reddit = praw.Reddit(config.user_agent)
self.reddit.set_oauth_app_info(client_id=config.client_id,
client_secret=config.client_secret,
redirect_uri=config.redirect_uri)
self.sub = self.reddit.get_subreddit(config.subreddit)
def authenticate(self):
access_information = self.reddit.refresh_access_information(self.refresh_token)
self.reddit.set_access_credentials(**access_information)
def get_posts(self):
return self.sub.get_hot(limit=self.post_limit)
def replied_to(self, obj):
replies = []
if type(obj) == praw.objects.Comment:
replies = obj.replies
elif type(obj) == praw.objects.Submission:
replies = obj.comments
else:
raise TypeException('Object not submission or comment')
return any(reply.author.name == config.username for reply in replies)
def own_comment(self, comment):
return comment.author.name == config.username
def parse_text(self, text):
tagRe = '\[\[([^\]]*)\]\]'
def parse_tag(tag_text):
subtags = {
'type': '\(\(([^\)]*)\)\)',
'faction': '\{\{([^\}]*)\}\}'
}
tag = {}
card = re.sub(tagRe, r'\1', tag_text)
card = reduce(lambda p, st: re.sub(subtags[st], '', p), subtags.keys(), card)
card = card.strip()
tag['card'] = card
for subtag_type, r in subtags.iteritems():
match = re.search(r, tag_text)
if match:
tag[subtag_type] = re.sub(r, r'\1', match.group()).strip()
else:
tag[subtag_type] = None
return tag
return [parse_tag(t) for t in re.findall(tagRe, text)]
def get_statline(self, card):
statline = {}
for stat in self.stats:
val = ''
if stat in card and card[stat] is not None:
val = str(card[stat])
if val is '' and 'ship' in card:
ship = self.api.get_ship(card['ship'])
val = str(ship[stat])
if val != '' and (stat != 'energy' or val != '0') and (stat != 'range' or val != '1-3'):
statline[stat] = val
return statline
def render_card_text(self, card):
replacements = {
'{STRONG}|{/STRONG}' : '**',
'{EM}|{/EM}' : '*',
'{BR}' : ''
}
text = card['text']
for pattern, repl in replacements.iteritems():
text = re.sub(pattern, repl, text)
return text
def render_card(self, card):
ret = '**Name:** ' + card['name'] + '\n\n'
statline = self.get_statline(card)
card_type = ''
if type(card['type']) is dict:
card_type = card['type']['name']
else:
card_type = self.api.get_upgrade_type(card['type'])['name']
ret += '**Type:** ' + card_type.capitalize() + ' '
for stat in self.stats:
if stat in statline and (stat != 'energy' or statline[stat] != '0'):
ret += '**' + stat.capitalize() + ':** ' + statline[stat] + ' '
ret += '\n\n'
ret += self.render_card_text(card)
return ret
def get_text(self, obj):
if type(obj) == praw.objects.Comment:
return obj.body
elif type(obj) == praw.objects.Submission:
return obj.selftext
else:
raise TypeException('Object not submission or comment')
def get_cards(self, tags):
cards = [self.api.get_card(t) for t in tags] #get cards
cards = [c for ca in cards for c in ca] #flatten card list
cards = [i for n,i in enumerate(cards) if i not in cards[n+1:]] #remove duplicates
cards = sorted(cards, key=lambda c: c['name']) #sort by name
return cards[:self.card_limit]
def build_comment(self, post):
comment = ''
tags = self.parse_text(self.get_text(post))
cards = self.get_cards(tags)
for index, card in enumerate(cards):
comment += self.render_card(card)
if index != len(cards) - 1:
comment += '\n\n \n\n---\n\n \n\n'
if comment != '':
comment += '\n\n \n\n Am I popping too much Glitterstim? Message /u/forkmonkey88'
return comment
def post_comment(self, obj, comment):
if self.debug:
print comment
return
submit = None
if type(obj) == praw.objects.Comment:
submit = obj.reply
elif type(obj) == praw.objects.Submission:
submit = obj.add_comment
else:
raise TypeException('Object not submission or comment')
submit(comment)
def process(self, obj):
try:
should_process = not self.replied_to(obj)
if type(obj) == praw.objects.Comment:
should_process = should_process and not self.own_comment(obj)
else:
for comment in praw.helpers.flatten_tree(obj.comments):
self.process(comment)
if should_process:
comment = self.build_comment(obj)
if comment:
self.post_comment(obj, comment)
except Exception:
print 'Unable to process: ' + str(obj)
if self.debug:
raise
def mash_go(self):
self.authenticate()
for post in self.get_posts():
self.process(post)
