Add civit module

Add civit.fetch function

shell/Handlers/CivitHandler.py

@@ -0,0 +1,111 @@
+import json
+
+from modules.civit.fetch import Fetch
+from shell.Handlers.ABS import Handler
+from modules.civit.client import Client
+
+class CivitHandler(Handler):
+
+
+
+    def __init__(self):
+        super().__init__()
+        self.forwarding_table: dict[str, Handler] = {
+            'fetch': FetchHandler(self)
+        }
+        self.handle_table: dict = {
+            'init': self._init,
+        }
+        self.client: Client | None = None
+
+
+    def _init(self, command: list[str], pos=0):
+        keys, args = self.parse_arguments(command[pos:], ['path', 'key'])
+        self._check_arg(keys, 'path')
+        self.client = Client(keys['path'], keys['key'])
+        self.succeed = True
+
+
+class FetchHandler(Handler):
+
+
+
+    def __init__(self, parent):
+        super().__init__()
+        self.parent: CivitHandler = parent
+        self.forwarding_table: dict[str, Handler] = {
+        }
+        self.handle_table: dict = {
+            'creators_raw': self._creators_raw,
+            'creators': self._creators,
+            'tags_raw': self._tags_raw,
+            'tags': self._tags,
+            'models': self._models,
+            'images': self._images,
+            'datamodel': self._datamodel,
+            'load': self._load,
+
+        }
+
+    def _load(self, command: list[str], pos=0):
+        keys, args = self.parse_arguments(command[pos:], ['entity'])
+        self._check_arg(keys, 'entity')
+
+        res = Fetch.load(self.parent.client, keys['entity'])
+        for r in res: print(r)
+        self.succeed = True
+
+
+
+    def _creators_raw(self, command: list[str], pos=0):
+        keys, args = self.parse_arguments(command[pos:], ['page', 'limit', 'query'])
+        res = self.parent.client.get_creators_raw(page=keys['page'], limit=keys['limit'], query=keys['query'])
+        print(res)
+        self.succeed = True
+
+    def _creators(self, command: list[str], pos=0):
+        res = Fetch.creators(self.parent.client)
+        for r in res: print(r)
+        self.succeed = True
+
+
+    def _tags_raw(self, command: list[str], pos=0):
+        keys, args = self.parse_arguments(command[pos:], ['page', 'limit', 'query'])
+        res = self.parent.client.get_tags_raw(page=keys['page'], limit=keys['limit'], query=keys['query'])
+        print(res)
+        self.succeed = True
+
+    def _tags(self, command: list[str], pos=0):
+        res = Fetch.tags(self.parent.client)
+        for r in res: print(r)
+        self.succeed = True
+
+    def _models(self, command: list[str], pos=0):
+        res = Fetch.models(self.parent.client)
+        for r in res: print(r)
+        self.succeed = True
+
+    def _images(self, command: list[str], pos=0):
+        keys, args = self.parse_arguments(command[pos:], ['start'])
+        res = Fetch.images(self.parent.client, start_with=keys['start'])
+        for r in res: print(r)
+        self.succeed = True
+
+    def _datamodel(self, command: list[str], pos=0):
+        entities = {
+            'creator': 'fetch_creators',
+            'tag': 'fetch_tags',
+            'model': 'fetch_models',
+            'image': 'fetch_images',
+        }
+        keys, args = self.parse_arguments(command[pos:], ['entity', 'top', 'dump'])
+        self._check_arg(keys, 'entity')
+        if keys['entity'] in entities:
+            res = Fetch.datamodel(self.parent.client, entities[keys['entity']], keys['top'] or 10)
+            print(json.dumps(res, indent=2, ensure_ascii=False))
+            if keys['dump']:
+                with open(keys['dump'], 'w') as f:
+                    json.dump(res, f, indent=2, ensure_ascii=False)
+
+
+        self.succeed = True

shell/Handlers/GlobalHandler.py

@@ -1,4 +1,5 @@
 from shell.Handlers.ABS import Handler
+from shell.Handlers.CivitHandler import CivitHandler
 from shell.Handlers.PythonappHandler import PythonappHandler
 from shell.Handlers.ModelSpaceHandler import ModelSpaceHandler
 
@@ -9,6 +10,7 @@ class GlobalHandler(Handler):
         self.forwarding_table: dict[str, Handler] = {
             'pythonapp': PythonappHandler(),
             'modelspace': ModelSpaceHandler(),
+            'civit': CivitHandler(),
         }
         self.handle_table: dict = {
             'tell': self._tell