Update login.py

Random-guy-on-internet · web-flow · commit 98a1a1bcb35b · 2021-01-08T17:49:21.000+03:30
diff --git a/Help_functions/login.py b/Help_functions/login.py
@@ -96,7 +96,7 @@ def get_nick_and_pw(server):
 session = ClientSession(headers=headers)
 
 
-async def get_content(link, data=None, login_first=False, return_url=False):
+async def get_content(link, data=None, login_first=False, return_url=False, first_run=False):
     """
     Return types:
     Method post -> respond url (unless fight.html in link -> tree and url)
@@ -112,7 +112,7 @@ async def get_content(link, data=None, login_first=False, return_url=False):
 
     server = link.split("#")[0].replace("http://", "https://").split("https://")[1].split(".e-sim.org")[0]
     method = "get" if not data and "fight.html" not in link and "medkit.html" not in link else "post"
-    if login_first:
+    if login_first or first_run:
         await login(server)
     async with session.get(link, cookies=cookies[server], headers=headers) if method == "get" else \
                session.post(link, cookies=cookies[server], headers=headers, data=data) as respond:
