def __init__(self, config_file='config/options.txt', user_file='config/users.txt'):
super().__init__()
self.config = Config(config_file)
self.kissAnime = KissDownloader()
self.users = []
if os.path.isfile(user_file):
with open(user_file, 'rb') as file:
self.users = pickle.load(file)
class AnimeBot(discord.Client):
def __init__(self, config_file='config/options.txt', user_file='config/users.txt'):
super().__init__()
self.config = Config(config_file)
self.kissAnime = KissDownloader()
self.users = []
if os.path.isfile(user_file):
with open(user_file, 'rb') as file:
self.users = pickle.load(file)
def run(self):
return super().run(self.config.username, self.config.password)
async def event_loop(self):
await asyncio.sleep(1)
while True:
for user in self.users:
try:
await self.check_for_user(user)
except Exception as e:
print(e)
print('Could not check updates for %s' % user.id)
await asyncio.sleep(300)
async def on_ready(self):
try:
win_unicode_console.enable()
except:
0
await self.change_status(game=discord.Game(name='with Eruru\'s tail'))
print('Connected!\n')
print('Username: %s' % self.user.name)
print('Bot ID: %s' % self.user.id)
print()
print('Command prefix is %s'% self.config.command_prefix)
print()
print('--Connected Servers List--')
if self.servers:
[print(s) for s in self.servers]
else:
print('No servers have been joined yet.')
print()
print('--Users Registered--')
if len(self.users) > 0:
for user in self.users:
print(user.id + ' - ' + user.kissUrl)
else:
print('No users have registered yet.')
print()
print('--Log--')
handler = getattr(self, 'event_loop', None)
await handler()
async def on_message(self, message):
if (message.channel.is_private) and (message.author != self.user) and (message.content.startswith(self.config.command_prefix)):
command = message.content[:message.content.find(' ')].replace(self.config.command_prefix, '')
data = message.content[message.content.find(' ')+1:]
if command == 'register':
if data.startswith('https://kissanime.to/MyList/'):
self.handle_register_user(message.author.id, data)
elif data.isdigit():
self.handle_register_user(message.author.id, 'https://kissanime.to/MyList/' + data)
if command == 'settts':
self.handle_set_tts(message.author.id, data)
def handle_register_user(self, userId, userUrl):
if self.get_user(userId) == 0:
user = User(userId, userUrl)
self.users.append(user)
print('Added user \'%s\' with url \'%s\'' % (userId, userUrl))
else:
self.get_user(userId).userUrl = userUrl
print('Updated bookmark url for user \'%s\'' % userId)
with open('config/users.txt', 'wb') as file:
pickle.dump(self.users, file)
def handle_set_tts(self, userId, channel):
user = self.get_user(userId)
if not user == 0:
user.ttsChannel = channel
print('Updated tts channel for \'%s\'' % userId)
with open('config/users.txt', 'wb') as file:
pickle.dump(self.users, file)
def get_user(self, userId):
for user in self.users:
if user.id == userId:
return user
return 0
async def check_for_user(self, user):
print('Checking bookmarks for \'%s\'...' % user.id)
cachedFilePath = 'cache/%s.dat' % user.id
kissDomain = 'https://kissanime.to'
colonId = '(*:*)'
# Download the users bookmark page
if os.path.isfile('bookmarkpage.html'):
with open('bookmarkpage.html', 'r') as file:
bookmarkPage = file.read()
else:
bookmarkPage = self.kissAnime.downloadPage(user.kissUrl).replace('\\r\\n', '')
#with open('bookmarkpage.html', 'w') as file:
# file.write(bookmarkPage)
# Turn the page into a list
newList = self.kiss_list_from_bookmarks(bookmarkPage)
# Load the old list from the file
oldList = {}
if os.path.isfile(cachedFilePath):
for line in open(cachedFilePath, 'r'):
try:
key, value = line.strip().split(': ')
key = key.replace(colonId, ':')
oldList[key] = tuple(value.replace('\'', '').replace('(', '').replace(')', '').split(', '))
except:
0 #best code evah
# Compare the lists and send messages
for key, newValue in newList.items():
try:
oldValue = oldList[key]
except:
oldValue = (newValue[0], '')
if oldValue[0] != newValue[0]:
await self.send_message(user.discordUser, 'The anime **%s** has just aired episode %s!\n%s' % (key, newValue[0], kissDomain + newValue[1]))
if (user.ttsChannel) and not (user.ttsChannel == ''):
channel = self.get_channel_class(user.ttsChannel.split('/')[0], user.ttsChannel.split('/')[1])
if not channel == 0:
message = 'The any may %s has just aired episode %s!' % (key.replace('.', '').replace('!', '').replace(',', '').replace(':', '').replace(';', ''), newValue[0])
await self.send_message(channel, message, tts=True)
# Save the new list into the file
with open(cachedFilePath, 'w') as file:
for key, value in newList.items():
file.write('%s: %s\n' % (key.replace(':', colonId), value))
print('Done checking bookmarks for \'%s\'!' % user.id)
def get_channel_class(self, serverId, channelId):
for server in self.servers:
if server.id == serverId:
for channel in server.channels:
if channel.id == channelId:
return channel
return 0
def kiss_list_from_bookmarks(self, content):
dataList = {}
table = content[content.find('<table class="listing">'):content.find('</table>')]
table = table[table.find('<tr class="trAnime'):table.find('</tbody>')]
rows = table.split('</tr>')
del rows[-1]
for row in rows:
try:
row += '</tr>'
soup = BeautifulSoup(row, 'html.parser')
key = soup.find_all('a')[1].string.strip()
episode = soup.find_all('a')[2].string.replace('Episode', '').replace('(', '[').replace(')', ']').strip()
link = soup.find_all('a')[1].get('href')
dataList[key] = (episode, link)
except:
0
return dataList
def kiss_latest_episode(self, content):
bowl = BeautifulSoup(content, 'html.parser').table
soup = BeautifulSoup(str(bowl), 'html.parser')
episode = soup.find_all('a')[0].string[-3:]
return episode
