Merge pull request TeamYukki#37 from azimazizov9150/patch-7

fannyxy · Jan 9, 2022 · 0f21334 · 0f21334
2 parents f99559f + 9c8cbab
commit 0f21334
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/gen_session.py b/gen_session.py
@@ -1,9 +1,20 @@
+from os import getenv
+
+from dotenv import load_dotenv
+
+load_dotenv()
+
+API_ID = int(getenv("API_ID", None))
+API_HASH = getenv("API_HASH", None)
+
 import asyncio
 
 from pyrogram import Client as c
 
-API_ID = input("\nEnter Your API_ID:\n > ")
-API_HASH = input("\nEnter Your API_HASH:\n > ")
+if not API_ID:
+    API_ID = input("\nEnter Your API_ID:\n > ")
+if not API_HASH:
+    API_HASH = input("\nEnter Your API_HASH:\n > ")
 
 print("\n\n Enter Phone number when asked.\n\n")