Update app.py

app.py

@@ -25,6 +25,9 @@ if MONGODB_URI:
         accounts_collection = db[COLLECTION_NAME]
         mongo_client.admin.command('ping')
         print("✅ MongoDB Connected!")
+        # Cek isi database
+        count = accounts_collection.count_documents({})
+        print(f"📊 Total dokumen di MongoDB: {count}")
     except ConnectionFailure as e:
         print(f"❌ MongoDB Connection Failed: {e}")
         mongo_client = None
@@ -60,20 +63,33 @@ def load_accounts_from_mongodb():
     accounts_dict = {}
     try:
         if mongo_client:
+            print("📥 Loading accounts from MongoDB...")
             cursor = accounts_collection.find({})
+            count = 0
             for doc in cursor:
+                count += 1
                 acc_id = doc.pop("_id")
                 doc["session"] = None
                 doc["csrf"] = None
                 doc["status"] = False
-                doc["otp_logs"] = []
+                doc["otp_logs"] = doc.get("otp_logs", [])
                 doc["sent_cache"] = []
                 doc["sms_cache"] = {}
                 doc["sms_counter"] = {}
                 doc["range_counter"] = {}
                 doc["last_cleanup"] = time.time()
                 accounts_dict[acc_id] = doc
+            
             print(f"📊 Loaded {len(accounts_dict)} accounts from MongoDB")
+            if count > 0:
+                print(f"✅ Data MongoDB berhasil dimuat: {count} akun")
+                # Tampilkan sample
+                for acc_id, acc in list(accounts_dict.items())[:2]:
+                    email_masked = mask_email(acc.get('username', 'Unknown'))
+                    log_count = len(acc.get('otp_logs', []))
+                    print(f"   - Akun: {email_masked} | OTP Logs: {log_count}")
+            else:
+                print("⚠️ Tidak ada data di MongoDB")
     except Exception as e:
         print(f"❌ Error load from MongoDB: {e}")
     return accounts_dict
@@ -81,22 +97,30 @@ def load_accounts_from_mongodb():
 def save_accounts_to_mongodb(accounts_dict):
     try:
         if mongo_client:
+            print("💾 Saving accounts to MongoDB...")
+            saved_count = 0
             for acc_id, acc in accounts_dict.items():
                 acc_copy = acc.copy()
                 acc_copy.pop("session", None)
                 acc_copy.pop("csrf", None)
-                if "otp_logs" in acc_copy and len(acc_copy["otp_logs"]) > 100:
-                    acc_copy["otp_logs"] = acc_copy["otp_logs"][:100]
                 acc_copy.pop("sms_cache", None)
                 acc_copy.pop("sms_counter", None)
                 acc_copy.pop("range_counter", None)
                 acc_copy.pop("last_cleanup", None)
-                accounts_collection.update_one(
+                
+                # Batasi logs
+                if "otp_logs" in acc_copy and len(acc_copy["otp_logs"]) > 100:
+                    acc_copy["otp_logs"] = acc_copy["otp_logs"][:100]
+                
+                result = accounts_collection.update_one(
                     {"_id": acc_id},
                     {"$set": acc_copy},
                     upsert=True
                 )
-            print(f"💾 Accounts saved to MongoDB")
+                if result.upserted_id or result.modified_count > 0:
+                    saved_count += 1
+            
+            print(f"💾 Saved {saved_count} accounts to MongoDB")
     except Exception as e:
         print(f"❌ Error save to MongoDB: {e}")
 
@@ -143,6 +167,7 @@ def save_accounts_to_file(accounts_dict):
     except Exception as e:
         print(f"❌ Error save to file: {e}")
 
+# Load accounts
 if mongo_client:
     accounts = load_accounts_from_mongodb()
 else:
@@ -831,6 +856,16 @@ def stream():
 def api_logs():
     return json.dumps(global_otp_logs[:100])
 
+@app.route('/api/stats')
+def api_stats():
+    return json.dumps({
+        "total_accounts": len(accounts),
+        "online_accounts": sum(1 for a in accounts.values() if a.get('status')),
+        "total_otp": len(global_otp_logs),
+        "mongo_connected": mongo_client is not None,
+        "mongo_docs": accounts_collection.count_documents({}) if mongo_client else 0
+    })
+
 def run_account_scraper(account_id):
     account = accounts.get(account_id)
     if not account:
@@ -935,12 +970,27 @@ def main():
     print("\n" + "="*70)
     print("  🔥 OTP MULTI ACCOUNT - FOURSTORE 🔥")
     print("="*70)
-    print(f"  ⚡ PORT: 7860")
-    print(f"  🌐 DOMAIN: {CUSTOM_DOMAIN}")
+    
+    # CEK MONGODB
     if mongo_client:
-        print("  📁 Data tersimpan di MongoDB")
+        print("  ✅ MongoDB: TERHUBUNG")
+        try:
+            count = accounts_collection.count_documents({})
+            print(f"  📊 Data di MongoDB: {count} akun")
+            
+            # Coba baca sample
+            sample = accounts_collection.find_one()
+            if sample:
+                print(f"  📝 Sample akun: {mask_email(sample.get('username', 'Unknown'))}")
+                log_count = len(sample.get('otp_logs', []))
+                print(f"  📝 OTP logs: {log_count}")
+        except Exception as e:
+            print(f"  ❌ Error cek MongoDB: {e}")
     else:
-        print("  📁 Data tersimpan di file accounts.json")
+        print("  ⚠️ MongoDB: TIDAK TERHUBUNG (pakai file)")
+    
+    print(f"  ⚡ PORT: 7860")
+    print(f"  🌐 DOMAIN: {CUSTOM_DOMAIN}")
     print("  📋 LOGGING: FULL DETAIL")
     print("  🔒 EMAIL SENSOR: AKTIF")
     print("  🤖 AUTO LOGIN: AKTIF")
@@ -950,6 +1000,8 @@ def main():
     print("  ⏰ 24 JAM MONITORING: AKTIF")
     print("="*70 + "\n")
     
+    # Muat ulang log dari akun yang ada
+    print("📥 Memuat ulang logs dari akun tersimpan...")
     for acc_id, acc in accounts.items():
         if "otp_logs" in acc and acc["otp_logs"]:
             for log in acc["otp_logs"]:
@@ -958,6 +1010,7 @@ def main():
     global_otp_logs.sort(key=lambda x: x.get('timestamp', 0), reverse=True)
     print(f"📊 Total logs dimuat: {len(global_otp_logs)}")
     
+    # Auto login untuk akun yang sudah login sebelumnya
     for acc_id, acc in accounts.items():
         if acc.get("status"):
             masked = mask_email(acc['username'])
@@ -991,9 +1044,15 @@ def main():
         time.sleep(300)
         if mongo_client:
             save_accounts_to_mongodb(accounts)
+            # Cek status MongoDB
+            try:
+                count = accounts_collection.count_documents({})
+                print(f"💾 Auto-save: {count} akun tersimpan - {get_wib_time_str()}")
+            except:
+                print(f"💾 Auto-save - {get_wib_time_str()}")
         else:
             save_accounts_to_file(accounts)
-        print(f"💾 Auto-save data - {get_wib_time_str()}")
+            print(f"💾 Auto-save data - {get_wib_time_str()}")
 
 if __name__ == "__main__":
     try: