diff --git a/app/core_utils.py b/app/core_utils.py old mode 100755 new mode 100644 index 77609ef6..b0f32d7a --- a/app/core_utils.py +++ b/app/core_utils.py @@ -57,13 +57,15 @@ def generate_embed( def gen_username(): - url = "https://generatorfun.com/code/model/generatorcontent.php?recordtable=generator&recordkey=960&gen=Y&itemnumber=1&randomoption=undefined&genimage=Yes&geneditor=No&nsfw=undefined&keyword=undefined&searchfilter=&searchfilterexclude=&tone=Normal&prefix=None&randomai=undefined" + from bs4 import BeautifulSoup - response = requests.get(url).text - # This is ugly as fuck but i didnt want to break out beautiful soup - return response.split( - "div id='gencont'>

" - )[1].split("

")[0] + url = "https://generatorfun.com/code/model/generatorcontent.php?recordtable=generator&recordkey=960&gen=Y&itemnumber=1&randomoption=undefined&genimage=Yes&geneditor=No&nsfw=undefined&keyword=undefined&searchfilter=&searchfilterexclude=&tone=Normal&prefix=None&randomai=undefined" + headers = {"User-Agent": "Mozilla/5.0 (Windows NT 6.1; Win64; x64)"} + response = requests.get(url, headers=headers).text + soup = BeautifulSoup(response, "html.parser") + + username = soup.find("div", {"id": "gencont"}).div.find("p").text + return username def waifu_pics(endpoint, nsfw=False):