Update app.py

app.py

@@ -19,14 +19,18 @@ MONGO_URI = os.environ.get("MONGGODB_URI")
 DB_NAME = "otp_bot"
 COLLECTION_NAME = "accounts"
 
-try:
-    mongo_client = MongoClient(MONGO_URI)
-    db = mongo_client[DB_NAME]
-    accounts_collection = db[COLLECTION_NAME]
-    mongo_client.admin.command('ping')
-    print("✅ MongoDB Connected!")
-except ConnectionFailure as e:
-    print(f"❌ MongoDB Connection Failed: {e}")
+if MONGO_URI:
+    try:
+        mongo_client = MongoClient(MONGO_URI)
+        db = mongo_client[DB_NAME]
+        accounts_collection = db[COLLECTION_NAME]
+        mongo_client.admin.command('ping')
+        print("✅ MongoDB Connected!")
+    except ConnectionFailure as e:
+        print(f"❌ MongoDB Connection Failed: {e}")
+        mongo_client = None
+else:
+    print("⚠️ MONGO_URI tidak di set, menggunakan penyimpanan lokal")
     mongo_client = None
 
 print = lambda *args, **kwargs: __builtins__.print(*args, **kwargs, flush=True)
@@ -39,6 +43,7 @@ GET_SMS_URL = f"{BASE}/portal/sms/received/getsms/number/sms"
 
 TELEGRAM_PROXY_URL = "https://danihitambangetjir.termai.cc/api/proxy"
 CUSTOM_DOMAIN = "https://fourstore-otp.hf.space"
+ACCOUNTS_FILE = "accounts.json"
 
 def mask_email(email):
     if not email or '@' not in email:
@@ -96,13 +101,57 @@ def save_accounts_to_mongodb(accounts_dict):
     except Exception as e:
         print(f"❌ Error save to MongoDB: {e}")
 
-accounts = load_accounts_from_mongodb()
+def load_accounts_from_file():
+    if os.path.exists(ACCOUNTS_FILE):
+        try:
+            with open(ACCOUNTS_FILE, 'r') as f:
+                data = json.load(f)
+                for acc_id in data:
+                    data[acc_id]["session"] = None
+                    data[acc_id]["csrf"] = None
+                    data[acc_id]["status"] = False
+                    data[acc_id]["otp_logs"] = []
+                    data[acc_id]["sent_cache"] = []
+                    data[acc_id]["sms_cache"] = {}
+                    data[acc_id]["sms_counter"] = {}
+                    data[acc_id]["range_counter"] = {}
+                    data[acc_id]["last_cleanup"] = time.time()
+                print(f"📊 Loaded {len(data)} accounts from file")
+                return data
+        except:
+            return {}
+    return {}
+
+def save_accounts_to_file(accounts_dict):
+    try:
+        accounts_to_save = {}
+        for acc_id, acc in accounts_dict.items():
+            acc_copy = acc.copy()
+            acc_copy.pop("session", None)
+            acc_copy.pop("csrf", None)
+            acc_copy.pop("otp_logs", None)
+            acc_copy.pop("sent_cache", None)
+            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_to_save[acc_id] = acc_copy
+        
+        with open(ACCOUNTS_FILE, 'w') as f:
+            json.dump(accounts_to_save, f, indent=2)
+        print(f"💾 Accounts saved to file")
+    except Exception as e:
+        print(f"❌ Error save to file: {e}")
+
+if mongo_client:
+    accounts = load_accounts_from_mongodb()
+else:
+    accounts = load_accounts_from_file()
 
 app = Flask('')
 app.secret_key = "fourstore-multi-account-secret"
 sse_clients = []
 
-# Decorator untuk cek owner via session
 def owner_required(f):
     @wraps(f)
     def decorated_function(*args, **kwargs):
@@ -123,7 +172,7 @@ def login_account(account_id, username, password, bot_token, chat_id, owner_id):
     try:
         masked = mask_email(username)
         print(f"\n{'='*60}")
-        print(f"🔐 PROSES LOGIN UNTUK: {masked} (ID: {account_id})")
+        print(f"🔐 PROSES LOGIN UNTUK: {masked} (ID: {account_id}) (Owner: {owner_id})")
         print(f"{'='*60}")
         
         session = httpx.Client(follow_redirects=True, timeout=30.0)
@@ -154,7 +203,11 @@ def login_account(account_id, username, password, bot_token, chat_id, owner_id):
             accounts[account_id]["chat_id"] = chat_id
             accounts[account_id]["owner_id"] = owner_id
             accounts[account_id]["last_login"] = time.time()
-            save_accounts_to_mongodb(accounts)
+            
+            if mongo_client:
+                save_accounts_to_mongodb(accounts)
+            else:
+                save_accounts_to_file(accounts)
             return True, "Login berhasil"
         else:
             return False, "Login gagal"
@@ -442,7 +495,6 @@ def logout_owner():
 def home():
     owner_id = flask_session.get('owner_id')
     
-    # Filter accounts berdasarkan owner_id
     user_accounts = {}
     for acc_id, acc in accounts.items():
         if acc.get("owner_id") == owner_id:
@@ -503,7 +555,6 @@ def home():
             body {{ font-family: 'Inter', sans-serif; background: #0a0c10; color: #e4e6eb; padding: 24px; }}
             .container {{ max-width: 1600px; margin: 0 auto; }}
             
-            /* Navbar */
             .navbar {{ background: #1a1f2c; border-radius: 16px; padding: 16px 24px; margin-bottom: 24px; display: flex; justify-content: space-between; align-items: center; border: 1px solid #2d3540; }}
             .nav-brand {{ font-size: 20px; font-weight: 700; color: #00f2fe; }}
             .nav-menu {{ display: flex; gap: 20px; align-items: center; }}
@@ -567,7 +618,6 @@ def home():
     </head>
     <body>
         <div class="container">
-            <!-- Navbar -->
             <div class="navbar">
                 <div class="nav-brand">OTP MULTI ACCOUNT</div>
                 <div class="nav-menu">
@@ -595,7 +645,7 @@ def home():
             </div>
             
             <div class="add-account-form">
-                <h3 style="margin-bottom: 15px;">➕ Tambah Akun Baru (Owner ID: {owner_id})</h3>
+                <h3 style="margin-bottom: 15px;">➕ Tambah Akun Baru</h3>
                 <form action="/add_account" method="POST" class="form-grid">
                     <input type="text" name="username" placeholder="Username/Email" class="form-input" required>
                     <input type="password" name="password" placeholder="Password" class="form-input" required>
@@ -606,7 +656,6 @@ def home():
                 </form>
             </div>
             
-            <!-- Preview Akun (3 teratas) -->
             <div style="margin-bottom: 20px; display: flex; justify-content: space-between; align-items: center;">
                 <h3>📋 Akun Terbaru</h3>
                 <a href="/accounts" class="btn btn-small">Lihat Semua Akun →</a>
@@ -742,7 +791,6 @@ def generate_preview_accounts(accounts_list):
 def accounts_page():
     owner_id = flask_session.get('owner_id')
     
-    # Filter accounts berdasarkan owner_id
     user_accounts = {}
     for acc_id, acc in accounts.items():
         if acc.get("owner_id") == owner_id:
@@ -954,10 +1002,13 @@ def add_account_route():
         "last_cleanup": time.time(),
         "created_at": time.time()
     }
-    save_accounts_to_mongodb(accounts)
+    
+    if mongo_client:
+        save_accounts_to_mongodb(accounts)
+    else:
+        save_accounts_to_file(accounts)
     print(f"✅ Akun ditambahkan: {masked}")
     
-    # OTOMATIS LOGIN SETELAH TAMBAH AKUN
     print(f"🔄 Mencoba login otomatis untuk {masked}...")
     success, msg = login_account(
         account_id, 
@@ -984,7 +1035,6 @@ def delete_account_route(account_id):
     owner_id = flask_session.get('owner_id')
     
     if account_id in accounts:
-        # Cek apakah akun ini milik owner yang sedang login
         if accounts[account_id].get("owner_id") != owner_id:
             return "Unauthorized", 403
             
@@ -992,8 +1042,11 @@ def delete_account_route(account_id):
         masked = mask_email(username)
         print(f"\n🗑️ HAPUS AKUN: {masked} (ID: {account_id})")
         del accounts[account_id]
+        
         if mongo_client:
             accounts_collection.delete_one({"_id": account_id})
+        else:
+            save_accounts_to_file(accounts)
         print(f"✅ Akun dihapus")
     
     return redirect(request.referrer or '/')
@@ -1005,7 +1058,6 @@ def login_account_route(account_id):
     print(f"\n🔔🔔🔔 LOGIN ACCOUNT DIPANGGIL: {account_id} 🔔🔔🔔")
     
     if account_id in accounts:
-        # Cek apakah akun ini milik owner yang sedang login
         if accounts[account_id].get("owner_id") != owner_id:
             return "Unauthorized", 403
             
@@ -1043,7 +1095,6 @@ def logout_account_route(account_id):
     owner_id = flask_session.get('owner_id')
     
     if account_id in accounts:
-        # Cek apakah akun ini milik owner yang sedang login
         if accounts[account_id].get("owner_id") != owner_id:
             return "Unauthorized", 403
             
@@ -1053,7 +1104,11 @@ def logout_account_route(account_id):
         accounts[account_id]["status"] = False
         accounts[account_id]["session"] = None
         accounts[account_id]["csrf"] = None
-        save_accounts_to_mongodb(accounts)
+        
+        if mongo_client:
+            save_accounts_to_mongodb(accounts)
+        else:
+            save_accounts_to_file(accounts)
         print(f"✅ Logout berhasil")
     
     return redirect(request.referrer or '/')
@@ -1170,14 +1225,16 @@ def main():
     print("  🔥 OTP MULTI ACCOUNT - FOURSTORE")
     print("  ⚡ PORT: 7860")
     print(f"  🌐 DOMAIN: {CUSTOM_DOMAIN}")
-    print("  📁 Data tersimpan di MongoDB")
+    if mongo_client:
+        print("  📁 Data tersimpan di MongoDB")
+    else:
+        print("  📁 Data tersimpan di file accounts.json")
     print("  📋 LOGGING: FULL DETAIL")
     print("  🔒 EMAIL SENSOR: AKTIF")
     print("  🤖 AUTO LOGIN: AKTIF (setelah tambah akun)")
-    print("  👑 OWNER SYSTEM: AKTIF")
+    print("  👑 OWNER SYSTEM: AKTIF (owner_id di input form)")
     print("="*60 + "\n")
     
-    # Auto login untuk akun yang sudah login sebelumnya
     for acc_id, acc in accounts.items():
         if acc.get("status"):
             masked = mask_email(acc['username'])
@@ -1197,7 +1254,10 @@ def main():
             else:
                 print(f"❌ {masked} offline: {msg}")
                 acc["status"] = False
-                save_accounts_to_mongodb(accounts)
+                if mongo_client:
+                    save_accounts_to_mongodb(accounts)
+                else:
+                    save_accounts_to_file(accounts)
     
     print("\n✅ BOT SIAP! Dashboard: https://fourstore-otp.hf.space")
     print("🔐 Login dengan Owner ID untuk mengakses")
@@ -1210,4 +1270,7 @@ if __name__ == "__main__":
         main()
     except KeyboardInterrupt:
         print("\n🛑 BOT STOPPED")
-        save_accounts_to_mongodb(accounts)
+        if mongo_client:
+            save_accounts_to_mongodb(accounts)
+        else:
+            save_accounts_to_file(accounts)