adrian/mutubot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

working a bit more

Browse Source
This commit is contained in:
Adrian 2024-11-07 19:48:17 +01:00
parent 8a3ded3673
commit 8f9e7d0bd6
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@ -65,16 +65,20 @@ for i in guilds:
from html.parser import HTMLParser from html.parser import HTMLParser
class TokenFinder(HTMLParser): class TokenFinder(HTMLParser):
def __init__(self): def __init__(self, admin_url):
self.token = None self.token = None
self.delete_links = []
self.admin_url = admin_url
super().__init__() super().__init__()
def handle_starttag(self, tag, attrs_tuple): def handle_starttag(self, tag, attrs_tuple):
if tag != 'input': if tag != 'input' and tag != 'a':
return return
attrs = dict(attrs_tuple) attrs = dict(attrs_tuple)
if tag == 'input' and 'type' in attrs and attrs['type'] == 'hidden' and 'name' in attrs and attrs['name'] == 'control' and 'value' in attrs : if tag == 'input' and 'type' in attrs and attrs['type'] == 'hidden' and 'name' in attrs and attrs['name'] == 'control' and 'value' in attrs :
self.token = attrs['value'] self.token = attrs['value']
elif tag == 'a' and 'href' in attrs and attrs['href'].startswith(self.admin_url + '/action/delete_column/') :
self.delete_links.append(attrs['href'])
def create_framavote (guild, names): def create_framavote (guild, names):