Create payments.py

chatbot/plugins/payments.py

@@ -0,0 +1,55 @@
+import time
+import os
+from pyrogram import *
+from pyrogram.types import *
+from database import db
+from datetime import datetime as dt, timedelta
+
+@Client.on_callback_query(filters.regex(r"^get_premium$"))
+async def get_premiums(client, callback):
+    keyboard = InlineKeyboardMarkup(
+        [
+            [
+                InlineKeyboardButton("$4.99 Monthly", callback_data="buy_premium_monthly"),
+                InlineKeyboardButton("$49.99 Yearly", callback_data="buy_premium_monthly"),
+            ],
+            [
+                InlineKeyboardButton(f"❌ Closed", callback_data="closedd")
+            ]
+        ]
+    )
+    return await callback.edit_message_reply_markup(reply_markup=keyboard)
+
+@Client.on_callback_query(filters.regex(r"^buy_premium_(monthly|yearly)$"))
+async def premium_purchase(client, callback):
+    plan = callback.matches[0].group(1)
+    user_id = callback.from_user.id
+    if user_id != 6477856957:
+        return await callback.answer("Only Developer test")
+
+    if plan == "monthly":
+        expiry = dt.now() + timedelta(days=30)
+        price = "$4.99"
+    else:
+        expiry = dt.now() + timedelta(days=365)
+        price = "$49.99"
+    
+    await db.user_premium.update_one(
+        {"user_id": user_id},
+        {
+            "$set": {
+                "is_premium": True,
+                "premium_expiry": expiry,
+                "subscription_plan": plan
+            },
+            "$inc": {"credits_used": -50}
+        },
+        upsert=True
+    )
+    await callback.message.edit_text(
+        f"💎 Premium {plan.capitalize()} Activated!\n"
+        f"∙ Expires: {expiry.strftime('%Y-%m-%d')}\n"
+        f"∙ Daily credits: 50 images\n\n"
+        f"Thanks for your support!",
+        reply_markup=None
+    )