Update app.py

app.py

@@ -16,6 +16,11 @@ from telethon.sessions import StringSession
 from telethon.tl import functions
 import re
 
+# ───── Utils ─────
+adjectives = ["Silent", "Crazy", "Golden", "Hidden", "Brave", "Fierce", "Mysterious", "Savage", "Royal", "Witty", "Legendary", "Epic", "Noble", "Clever", "Loyal", "Wild", "Electric", "Fearless", "Crimson", "Frozen", "Charming", "Rebel", "Turbo", "Lone", "Radiant", "Raging", "Dusty", "Majestic", "Grumpy", "Shadow"]
+nouns = ["Warriors", "Coders", "Friends", "Legends", "Pirates", "Tigers", "Wolves", "Hackers", "Ninjas", "Riders", "Hunters", "Kings", "Queens", "Giants", "Lords", "Champions", "Stormers", "Creators", "Ghosts", "Wizards", "Knights", "Samurais", "Explorers", "Titans", "Outlaws", "Rebels", "Squad", "Troopers", "Dreamers", "Avengers"]
+
+
 logging.basicConfig(
     level=logging.INFO,
     format="%(asctime)s | %(message)s",
@@ -33,60 +38,56 @@ ADMIN_ID = int(os.getenv("ADMIN_ID", 6574063018))
 client = MongoClient(MONGO_URI)
 db = client["telethon_bots"]
 accounts = db["accounts"]
-bot = TelegramClient("main_bot", API_ID, API_HASH)
-userbots = {}
-
-scheduler = AsyncIOScheduler(timezone=pytz.timezone("Asia/Kolkata"))
+plans = db["plans"]  # new collection to track next day schedule
 
-# ───── Reset + Plan ─────
-def reset_and_plan():
-    accounts.update_many({}, {"$set": {"created": 0, "schedule_times": []}})
-    all_accounts = list(accounts.find())
+def random_group_name():
+    return f"{random.choice(adjectives)} {random.choice(nouns)}"
 
-    for acc in all_accounts:
+def create_tomorrow_plan():
+    tz = pytz.timezone("Asia/Kolkata")
+    tomorrow = (datetime.now(tz) + timedelta(days=1)).replace(hour=0, minute=0, second=0, microsecond=0)
+    plans.delete_many({})  # clear old plans
+    for acc in accounts.find():
         limit = random.randint(3, 5)
-        times = []
+        times = set()
         while len(times) < limit:
             h, m = random.randint(0, 23), random.randint(0, 59)
-            dt = datetime.now(pytz.timezone("Asia/Kolkata")).replace(hour=h, minute=m, second=0, microsecond=0)
-            if dt not in times:
-                times.append(dt)
-        times.sort()
-        accounts.update_one({"_id": acc["_id"]}, {"$set": {"schedule_times": times}})
-
-        if acc["user_id"] in userbots:
-            for run_time in times:
-                scheduler.add_job(auto_create_groups, 'date', run_date=run_time, args=[userbots[acc["user_id"]], acc])
+            t = tomorrow + timedelta(hours=h, minutes=m)
+            times.add(t)
+        schedule_times = sorted(list(times))
+        plans.insert_one({"user_id": acc["user_id"], "times": schedule_times, "created": 0})
+    log.info("📅 Tomorrow's plan created.")
+
+def apply_today_plan():
+    tz = pytz.timezone("Asia/Kolkata")
+    now = datetime.now(tz)
+    plans_today = list(plans.find())
+    for plan in plans_today:
+        acc = accounts.find_one({"user_id": plan["user_id"]})
+        if not acc or plan.get("created", 0) >= len(plan["times"]):
+            continue
+        for run_time in plan["times"]:
+            if run_time.date() == now.date():
+                scheduler.add_job(auto_create_groups, 'date', run_date=run_time, args=[userbots[plan["user_id"]], acc, plan])
 
-    asyncio.create_task(bot.send_message(ADMIN_ID, "🔄 Daily limits & schedules updated."))
-    log.info("🔄 Daily plan and reset done.")
+def reset_and_plan():
+    tz = pytz.timezone("Asia/Kolkata")
+    plans.delete_many({})
+    create_tomorrow_plan()
+    apply_today_plan()
+    log.info("🔄 New day plan set.")
+    asyncio.create_task(bot.send_message(ADMIN_ID, "🗓️ Daily schedule reset and updated."))
 
 scheduler.add_job(reset_and_plan, CronTrigger(hour=0, minute=0))
 
-# ───── Utils ─────
-adjectives = ["Silent", "Crazy", "Golden", "Hidden", "Brave", "Fierce", "Mysterious", "Savage", "Royal", "Witty", "Legendary", "Epic", "Noble", "Clever", "Loyal", "Wild", "Electric", "Fearless", "Crimson", "Frozen", "Charming", "Rebel", "Turbo", "Lone", "Radiant", "Raging", "Dusty", "Majestic", "Grumpy", "Shadow"]
-nouns = ["Warriors", "Coders", "Friends", "Legends", "Pirates", "Tigers", "Wolves", "Hackers", "Ninjas", "Riders", "Hunters", "Kings", "Queens", "Giants", "Lords", "Champions", "Stormers", "Creators", "Ghosts", "Wizards", "Knights", "Samurais", "Explorers", "Titans", "Outlaws", "Rebels", "Squad", "Troopers", "Dreamers", "Avengers"]
-
-def random_group_name():
-    return f"{random.choice(adjectives)} {random.choice(nouns)}"
-
-def schedule_random_group_creation(userbot, acc_doc):
-    if acc_doc.get("schedule_times"):
-        for run_time in acc_doc["schedule_times"]:
-            scheduler.add_job(auto_create_groups, 'date', run_date=run_time, args=[userbot, acc_doc])
+async def auto_create_groups(userbot, acc_doc, plan_doc):
+    if plan_doc["created"] >= len(plan_doc["times"]):
+        await bot.send_message(ADMIN_ID, f"❌ Group creation skipped: Limit exceeded for {acc_doc['name']}")
+        return
 
-# ───── Group Creation ──���──
-async def auto_create_groups(userbot, acc_doc):
     try:
-        if acc_doc["created"] >= len(acc_doc.get("schedule_times", [])):
-            return
-
         title = random_group_name()
-        result = await userbot(functions.channels.CreateChannelRequest(
-            title=title,
-            about=",",
-            megagroup=True
-        ))
+        result = await userbot(functions.channels.CreateChannelRequest(title=title, about=",", megagroup=True))
         group = result.chats[0].id
 
         for _ in range(4):
@@ -96,25 +97,16 @@ async def auto_create_groups(userbot, acc_doc):
         try:
             stickers = ["CAACAgUAAxkBAAEKZStlTxkC-cvQJwABvHoUjRacAVIHCFcAAgECAALaTqVVCpx_Lc2ZoLguBA"]
             await userbot.send_file(group, random.choice(stickers))
-        except Exception:
-            pass
+        except: pass
 
-        accounts.update_one(
-            {"_id": acc_doc["_id"]},
-            {"$inc": {"created": 1}, "$set": {"last_created": datetime.now(pytz.timezone("Asia/Kolkata"))}}
-        )
+        plans.update_one({"user_id": acc_doc["user_id"]}, {"$inc": {"created": 1}})
 
         info = await userbot.get_me()
-        await bot.send_message(
-            ADMIN_ID,
-            f"✅ Created 1 group\n👤 {info.first_name}\n🔗 @{info.username or 'N/A'}\n📊 Total today: {acc_doc['created'] + 1}"
-        )
-        log.info(f"👥 1 group created for {acc_doc['name']}")
+        await bot.send_message(ADMIN_ID, f"✅ Created 1 group\n👤 {info.first_name}\n🔗 @{info.username or 'N/A'}\n📊 Created today: {plan_doc['created'] + 1}")
     except Exception as e:
         await bot.send_message(ADMIN_ID, f"❌ Group creation failed for {acc_doc['name']}: {e}")
         log.error(f"❌ Group creation failed: {e}")
 
-# ───── Userbot Setup ─────
 async def start_userbot(session_str, owner_id):
     try:
         userbot = TelegramClient(StringSession(session_str), API_ID, API_HASH)
@@ -133,14 +125,12 @@ async def start_userbot(session_str, owner_id):
         }
         accounts.insert_one(doc)
         userbots[info.id] = userbot
-        schedule_random_group_creation(userbot, doc)
         asyncio.create_task(userbot.run_until_disconnected())
         return info
     except Exception as e:
         log.info(f"❌ Failed to add account: {e}")
         return None
 
-# ───── Load All ─────
 async def load_all_userbots():
     log.info("🔄 Loading all stored userbots...")
     for acc in accounts.find():
@@ -148,13 +138,13 @@ async def load_all_userbots():
             ub = TelegramClient(StringSession(acc["session"]), API_ID, API_HASH)
             await ub.start()
             userbots[acc["user_id"]] = ub
-            schedule_random_group_creation(ub, acc)
             asyncio.create_task(ub.run_until_disconnected())
             log.info(f"✅ Loaded: {acc['name']} (@{acc.get('username', 'N/A')})")
         except Exception as e:
             await bot.send_message(ADMIN_ID, f"❌ Failed to start {acc['name']}")
             log.info(f"❌ Failed to start {acc['name']}: {e}")
 
+
 # ───── Bot Commands ─────
 @bot.on(events.NewMessage(pattern="/start"))
 async def start_cmd(event):