async def scrawl(self, threads=5): logger.log('Scrawling Trackemon..', 'green') await self.client.wait_until_ready() # get arrays channels id need to post shout_out_channels = [] for server in self.client.servers: for channel in server.channels: if channel.name in self.config.get('scrawl_channels', []): shout_out_channels.append(discord.Object(channel.id)) if len(shout_out_channels) == 0: raise Exception("No channel to shout out!") while not self.client.is_closed: logger.log('Scrawling Trackemon..', 'green') self._retrieve_session_id() # use multiprocessing if 'pokemons' in self.config.get('scrawl_trackemon'): pokemon_names = self.config.get('scrawl_trackemon')['pokemons'] pool = ThreadPool(threads) messages = pool.starmap(self.scrawl_trackemon, zip( pokemon_names, itertools.repeat(self.session_id))) for message in messages: if len(message): for channel in shout_out_channels: await self.client.send_message(channel, message) # increase delay to finish task await asyncio.sleep(self.config.get('delay_scrawl', 300))
async def announcement(self): logger.log('Posting announcements..') announcement = self.config.get('announcement') message = announcement.get('message', None) delay = announcement.get('delay', 300) if not message: logger.log('Please config your announcement message', 'red') return await self.client.wait_until_ready() # get arrays channels id need to post announcement_channels = [] for server in self.client.servers: for channel in server.channels: if channel.name in self.config.get('channels', []): announcement_channels.append(discord.Object(channel.id)) while not self.client.is_closed: for channel in announcement_channels: await self.client.send_message(channel, message) await asyncio.sleep(delay)
def need_to_delete(self, message): except_roles = self.config.get('except_roles', []) # Do nothing if bot is sending message if message.author == self.client.user: return False """ Ignore if message from except roles """ for role in message.author.roles: if role.name in except_roles: return False """ Check if message contains lat/long & pokemon name """ pattern = "^(\-?\d+(\.\d+)?),\s*(\-?\d+(\.\d+)?)(?=.*(Bulbasaur|bulbasaur|Ivysaur|ivysaur|Venusaur|venusaur|Charmander|charmander|Charmeleon|charmeleon|Charizard|charizard|Squirtle|squirtle|Wartortle|wartortle|Blastoise|blastoise|Caterpie|caterpie|Metapod|metapod|Butterfree|butterfree|Weedle|weedle|Kakuna|kakuna|Beedrill|beedrill|Pidgey|pidgey|Pidgeotto|pidgeotto|Pidgeot|pidgeot|Rattata|rattata|Raticate|raticate|Spearow|spearow|Fearow|fearow|Ekans|ekans|Arbok|arbok|Pikachu|pikachu|Raichu|raichu|Sandshrew|sandshrew|Sandslash|sandslash|Nidoran F|nidoran f|Nidorina|nidorina|Nidoqueen|nidoqueen|Nidoran M|nidoran m|Nidorino|nidorino|Nidoking|nidoking|Clefairy|clefairy|Clefable|clefable|Vulpix|vulpix|Ninetales|ninetales|Jigglypuff|jigglypuff|Wigglytuff|wigglytuff|Zubat|zubat|Golbat|golbat|Oddish|oddish|Gloom|gloom|Vileplume|vileplume|Paras|paras|Parasect|parasect|Venonat|venonat|Venomoth|venomoth|Diglett|diglett|Dugtrio|dugtrio|Meowth|meowth|Persian|persian|Psyduck|psyduck|Golduck|golduck|Mankey|mankey|Primeape|primeape|Growlithe|growlithe|Arcanine|arcanine|Poliwag|poliwag|Poliwhirl|poliwhirl|Poliwrath|poliwrath|Abra|abra|Kadabra|kadabra|Alakazam|alakazam|Machop|machop|Machoke|machoke|Machamp|machamp|Bellsprout|bellsprout|Weepinbell|weepinbell|Victreebel|victreebel|Tentacool|tentacool|Tentacruel|tentacruel|Geodude|geodude|Graveler|graveler|Golem|golem|Ponyta|ponyta|Rapidash|rapidash|Slowpoke|slowpoke|Slowbro|slowbro|Magnemite|magnemite|Magneton|magneton|Farfetchd|farfetchd|Doduo|doduo|Dodrio|dodrio|Seel|seel|Dewgong|dewgong|Grimer|grimer|Muk|muk|Shellder|shellder|Cloyster|cloyster|Gastly|gastly|Haunter|haunter|Gengar|gengar|Onix|onix|Drowzee|drowzee|Hypno|hypno|Krabby|krabby|Kingler|kingler|Voltorb|voltorb|Electrode|electrode|Exeggcute|exeggcute|Exeggutor|exeggutor|Cubone|cubone|Marowak|marowak|Hitmonlee|hitmonlee|Hitmonchan|hitmonchan|Lickitung|lickitung|Koffing|koffing|Weezing|weezing|Rhyhorn|rhyhorn|Rhydon|rhydon|Chansey|chansey|Tangela|tangela|Kangaskhan|kangaskhan|Horsea|horsea|Seadra|seadra|Goldeen|goldeen|Seaking|seaking|Staryu|staryu|Starmie|starmie|MrMime|mrmime|Scyther|scyther|Jynx|jynx|Electabuzz|electabuzz|Magmar|magmar|Pinsir|pinsir|Tauros|tauros|Magikarp|magikarp|Gyarados|gyarados|Lapras|lapras|Ditto|ditto|Eevee|eevee|Vaporeon|vaporeon|Jolteon|jolteon|Flareon|flareon|Porygon|porygon|Omanyte|omanyte|Omastar|omastar|Kabuto|kabuto|Kabutops|kabutops|Aerodactyl|aerodactyl|Snorlax|snorlax|Articuno|articuno|Zapdos|zapdos|Moltres|moltres|Dratini|dratini|Dragonair|dragonair|Dragonite|dragonite|Mewtwo|mewtwo|Mew|mew)).*$" reverse_pattern = "^(Bulbasaur|bulbasaur|Ivysaur|ivysaur|Venusaur|venusaur|Charmander|charmander|Charmeleon|charmeleon|Charizard|charizard|Squirtle|squirtle|Wartortle|wartortle|Blastoise|blastoise|Caterpie|caterpie|Metapod|metapod|Butterfree|butterfree|Weedle|weedle|Kakuna|kakuna|Beedrill|beedrill|Pidgey|pidgey|Pidgeotto|pidgeotto|Pidgeot|pidgeot|Rattata|rattata|Raticate|raticate|Spearow|spearow|Fearow|fearow|Ekans|ekans|Arbok|arbok|Pikachu|pikachu|Raichu|raichu|Sandshrew|sandshrew|Sandslash|sandslash|Nidoran F|nidoran f|Nidorina|nidorina|Nidoqueen|nidoqueen|Nidoran M|nidoran m|Nidorino|nidorino|Nidoking|nidoking|Clefairy|clefairy|Clefable|clefable|Vulpix|vulpix|Ninetales|ninetales|Jigglypuff|jigglypuff|Wigglytuff|wigglytuff|Zubat|zubat|Golbat|golbat|Oddish|oddish|Gloom|gloom|Vileplume|vileplume|Paras|paras|Parasect|parasect|Venonat|venonat|Venomoth|venomoth|Diglett|diglett|Dugtrio|dugtrio|Meowth|meowth|Persian|persian|Psyduck|psyduck|Golduck|golduck|Mankey|mankey|Primeape|primeape|Growlithe|growlithe|Arcanine|arcanine|Poliwag|poliwag|Poliwhirl|poliwhirl|Poliwrath|poliwrath|Abra|abra|Kadabra|kadabra|Alakazam|alakazam|Machop|machop|Machoke|machoke|Machamp|machamp|Bellsprout|bellsprout|Weepinbell|weepinbell|Victreebel|victreebel|Tentacool|tentacool|Tentacruel|tentacruel|Geodude|geodude|Graveler|graveler|Golem|golem|Ponyta|ponyta|Rapidash|rapidash|Slowpoke|slowpoke|Slowbro|slowbro|Magnemite|magnemite|Magneton|magneton|Farfetchd|farfetchd|Doduo|doduo|Dodrio|dodrio|Seel|seel|Dewgong|dewgong|Grimer|grimer|Muk|muk|Shellder|shellder|Cloyster|cloyster|Gastly|gastly|Haunter|haunter|Gengar|gengar|Onix|onix|Drowzee|drowzee|Hypno|hypno|Krabby|krabby|Kingler|kingler|Voltorb|voltorb|Electrode|electrode|Exeggcute|exeggcute|Exeggutor|exeggutor|Cubone|cubone|Marowak|marowak|Hitmonlee|hitmonlee|Hitmonchan|hitmonchan|Lickitung|lickitung|Koffing|koffing|Weezing|weezing|Rhyhorn|rhyhorn|Rhydon|rhydon|Chansey|chansey|Tangela|tangela|Kangaskhan|kangaskhan|Horsea|horsea|Seadra|seadra|Goldeen|goldeen|Seaking|seaking|Staryu|staryu|Starmie|starmie|MrMime|mrmime|Scyther|scyther|Jynx|jynx|Electabuzz|electabuzz|Magmar|magmar|Pinsir|pinsir|Tauros|tauros|Magikarp|magikarp|Gyarados|gyarados|Lapras|lapras|Ditto|ditto|Eevee|eevee|Vaporeon|vaporeon|Jolteon|jolteon|Flareon|flareon|Porygon|porygon|Omanyte|omanyte|Omastar|omastar|Kabuto|kabuto|Kabutops|kabutops|Aerodactyl|aerodactyl|Snorlax|snorlax|Articuno|articuno|Zapdos|zapdos|Moltres|moltres|Dratini|dratini|Dragonair|dragonair|Dragonite|dragonite|Mewtwo|mewtwo|Mew|mew)(?=.*(\-?\d+(\.\d+)?),\s*(\-?\d+(\.\d+)?)).*$" # Manage channel rare_spottings only if message.channel.name in self.config.get('channels', []): if not (re.match(pattern, message.content) or re.match( reverse_pattern, message.content)) or self._is_blacklisted( message.content): if self._is_whitelisted(message.content): return False # Log & print out log_message = 'Message has been deleted: {} - Author: {}'.format( message.content, message.author.name) logger.log(log_message) logging.warning(log_message) # Delete message return True
def _retrieve_session_id(self, delay=5): if not self.session_id: # get trackemon session session_id = self._get_trackemon_session() if not session_id: logger.log( 'Cannot retrieve session id for Trackemon.. Trying again', 'red') time.sleep(delay) return self._retrieve_session_id() else: self.session_id = session_id
def _retrieve_session_id(self, delay=5): if not self.session_id: # get trackemon session session_id = self._get_trackemon_session() if not session_id: logger.log( 'Cannot retrieve session id for Trackemon.. Trying again', 'red') time.sleep(delay) return self._retrieve_session_id() else: self.session_id = session_id
async def on_ready(): """ Connected to bot """ logger.log('Logged in as') logger.log(client.user.name) logger.log(client.user.id) logger.log('------')
async def on_ready(): """ Connected to bot """ logger.log('Logged in as') logger.log(client.user.name) logger.log(client.user.id) logger.log('------')
def scrawl_trackemon(self, pokemon_name, session_id): """ Scrawl api from trackemon.com """ session = self.bot.session api_root = 'http://www.trackemon.com/' # get pokemon id try: pokemon_id = getattr(Pokemon, pokemon_name.upper()).value except AttributeError: logger.log('Wrong pokemon name', 'red') sys.exit() api_endpoint = '{}fetch/rare?pokedexTypeId={}&sessionId={}' endpoint = api_endpoint.format(api_root, pokemon_id, session_id) logger.log(endpoint) try: response = session.get(endpoint) data = response.json() except: logger.log('Api is busy') return '' if not data: logger.log('There is no {}'.format(pokemon_name), 'yellow') else: logger.log('There are {} {}(s)'.format( len(data), pokemon_name), 'blue') message = '' for pokemon in data: expire_seconds = int(pokemon['expirationTime']) - time.time() # format message message += '`{}` {},{} - Expires in: {:.0f} seconds \n'.format( self._get_pokemon_name(pokemon['pokedexTypeId']), pokemon['latitude'], pokemon['longitude'], expire_seconds ) return message
def _get_trackemon_session(self): session = self.bot.session # get session_id by fetching root first_response = session.get('http://www.trackemon.com/') pattern = re.compile(r'sessionId\s=\s\'\w+\'') session_id_string = re.search( pattern, first_response.content.decode('utf-8')) if session_id_string: # Found string, split session id session_id = session_id_string.group().replace( 'sessionId = ', '').replace('\'', '') else: logger.log( 'Can\'t retrieve session id for api. Waiting for next scrawl..') # init session again return None return session_id
def scrawl_trackemon(self, pokemon_name, session_id): """ Scrawl api from trackemon.com """ session = self.bot.session api_root = 'http://www.trackemon.com/' # get pokemon id try: pokemon_id = getattr(Pokemon, pokemon_name.upper()).value except AttributeError: logger.log('Wrong pokemon name', 'red') sys.exit() api_endpoint = '{}fetch/rare?pokedexTypeId={}&sessionId={}' endpoint = api_endpoint.format(api_root, pokemon_id, session_id) logger.log(endpoint) try: response = session.get(endpoint) data = response.json() except: logger.log('Api is busy') return '' if not data: logger.log('There is no {}'.format(pokemon_name), 'yellow') else: logger.log('There are {} {}(s)'.format(len(data), pokemon_name), 'blue') message = '' for pokemon in data: expire_seconds = int(pokemon['expirationTime']) - time.time() # format message message += '`{}` {},{} - Expires in: {:.0f} seconds \n'.format( self._get_pokemon_name(pokemon['pokedexTypeId']), pokemon['latitude'], pokemon['longitude'], expire_seconds) return message
def _get_trackemon_session(self): session = self.bot.session # get session_id by fetching root first_response = session.get('http://www.trackemon.com/') pattern = re.compile(r'sessionId\s=\s\'\w+\'') session_id_string = re.search(pattern, first_response.content.decode('utf-8')) if session_id_string: # Found string, split session id session_id = session_id_string.group().replace('sessionId = ', '').replace( '\'', '') else: logger.log( 'Can\'t retrieve session id for api. Waiting for next scrawl..' ) # init session again return None return session_id
async def scrawl(self): await self.client.wait_until_ready() # get arrays channels id need to post shout_out_channels = [] for server in self.client.servers: for channel in server.channels: if channel.name in self.config.get('scrawl_channels', []): shout_out_channels.append(discord.Object(channel.id)) if len(shout_out_channels) == 0: raise Exception("No channel to shout out!") while not self.client.is_closed: logger.log('Scrawling PokeSnipers..', 'green') # Use scraper to bypass cloudflare ddos protection scraper = cfscrape.create_scraper() json_string = scraper.get( 'http://pokesnipers.com/api/v1/pokemon.json').content data = json.loads(json_string.decode('utf-8')) logger.log('Got {} pokemons from PokeSnipers'.format( len(data.get('results'))), 'blue') message = '' for pokemon in data.get('results'): until = time.mktime(time.strptime(pokemon.get('until'), '%Y-%m-%dT%H:%M:%S.%fZ')) # use utc timezone for pokesnipers expire_seconds = until - datetime.datetime.utcnow().timestamp() message += "`{}` {} - Expires in: {:.0f} seconds\n".format(pokemon.get('name'), pokemon.get('coords'), expire_seconds) if len(message): for channel in shout_out_channels: await self.client.send_message(channel, message) await asyncio.sleep(self.config.get('delay_scrawl', 300))
async def scrawl(self, threads=5): logger.log('Scrawling Trackemon..', 'green') await self.client.wait_until_ready() # get arrays channels id need to post shout_out_channels = [] for server in self.client.servers: for channel in server.channels: if channel.name in self.config.get('scrawl_channels', []): shout_out_channels.append(discord.Object(channel.id)) if len(shout_out_channels) == 0: raise Exception("No channel to shout out!") while not self.client.is_closed: logger.log('Scrawling Trackemon..', 'green') self._retrieve_session_id() # use multiprocessing if 'pokemons' in self.config.get('scrawl_trackemon'): pokemon_names = self.config.get('scrawl_trackemon')['pokemons'] pool = ThreadPool(threads) messages = pool.starmap( self.scrawl_trackemon, zip(pokemon_names, itertools.repeat(self.session_id))) for message in messages: if len(message): for channel in shout_out_channels: await self.client.send_message(channel, message) # increase delay to finish task await asyncio.sleep(self.config.get('delay_scrawl', 300))