WIP: #50

classes/api_client.py

@@ -0,0 +1,21 @@
+from garbage_api_client import Client
+from garbage_api_client.api.default.entry_find_locations_location_entries_get import (
+    asyncio as entry_find,
+)
+from garbage_api_client.api.default.location_find_locations_get import (
+    asyncio as location_find,
+)
+from garbage_api_client.api.default.location_get_locations_id_get import (
+    asyncio as location_get,
+)
+
+
+class APIClient(Client):
+    async def entry_find(self, *args, **kwargs):
+        await entry_find(client=self, *args, **kwargs)
+
+    async def location_find(self, *args, **kwargs):
+        await location_find(client=self, *args, **kwargs)
+
+    async def location_get(self, *args, **kwargs):
+        await location_get(client=self, *args, **kwargs)

classes/pyroclient.py

@@ -8,6 +8,7 @@ from libbot.pyrogram.classes import PyroClient as LibPyroClient
 from pymongo import ASCENDING, GEOSPHERE, TEXT
 from pyrogram.types import User
 
+from classes.api_client import APIClient
 from classes.location import Location
 from classes.pyrouser import PyroUser
 from classes.updater import Updater
@@ -25,6 +26,7 @@ class PyroClient(LibPyroClient):
 
         self.updater = Updater(ClientSession())
         self.contexts = []
+        self.api_client = APIClient()
 
         if self.scheduler is not None:
             self.scheduler.add_job(

config_example.json

@@ -9,6 +9,9 @@
         "max_concurrent_transmissions": 1,
         "scoped_commands": true
     },
+    "api": {
+        "url": "https://api.garbagebot.eu"
+    },
     "database": {
         "user": null,
         "password": null,
@@ -16,13 +19,6 @@
         "port": 27017,
         "name": "garbage_bot"
     },
-    "database_api": {
-        "user": null,
-        "password": null,
-        "host": "127.0.0.1",
-        "port": 27017,
-        "name": "garbage_reminder"
-    },
     "search": {
         "radius": 0.1
     },

migrations/202405270030.py

@@ -0,0 +1,21 @@
+from libbot import sync
+from mongodb_migrations.base import BaseMigration
+
+
+class Migration(BaseMigration):
+    def upgrade(self):
+        sync.config_set("api", {"url": "https://api.garbagebot.eu"})
+        sync.config_delete("database_api", missing_ok=True)
+
+    def downgrade(self):
+        sync.config_delete("api", missing_ok=True)
+        sync.config_set(
+            "database_api",
+            {
+                "user": None,
+                "password": None,
+                "host": "127.0.0.1",
+                "port": 27017,
+                "name": "garbage_reminder",
+            },
+        )

requirements.txt

@@ -9,3 +9,5 @@ uvloop==0.19.0
 async_pymongo==0.1.4
 libbot[speed,pyrogram]==3.2.2
 pykeyboard==0.1.7
+--extra-index-url https://git.end-play.xyz/api/packages/GarbageReminder/pypi/simple
+garbageapi-client==0.1.0