edi-core 0.3.0__py3-none-any.whl

edi_core/ai/writesonic/__init__.py

@@ -0,0 +1,8 @@
+WRITESONIC_BASE_URL = 'https://api.writesonic.com'
+CHATSONIC_URL_PATH = '/v2/business/content/chatsonic'
+WRITESONIC_ENGINE_PREMIUM = 'premium'
+WRITESONIC_ENGINE_SUPERIUM = 'superium'
+WRITESONIC_ENGINES = (WRITESONIC_ENGINE_PREMIUM, WRITESONIC_ENGINE_SUPERIUM)
+WRITESONIC_LANGUAGE_ENGLISH = 'en'
+WRITESONIC_LANGUAGE_CHINESE = 'zh'
+WRITESONIC_LANGUAGES = (WRITESONIC_LANGUAGE_ENGLISH, WRITESONIC_LANGUAGE_CHINESE)

edi_core/ai/writesonic/chatsonic.py

@@ -0,0 +1,108 @@
+import json
+import os
+from dataclasses import dataclass, field
+from typing import List
+
+import requests
+from edi_core.ai.writesonic import *
+
+
+@dataclass(kw_only=True)
+class ChatSonicConfig:
+    api_key: str = os.environ.get("SONIC_KEY")
+    base_url: str = WRITESONIC_BASE_URL
+    path: str = CHATSONIC_URL_PATH
+    engine: str = WRITESONIC_ENGINE_PREMIUM
+    language: str = WRITESONIC_LANGUAGE_ENGLISH
+    enable_memory: bool = False
+    enable_google_results: bool = False
+
+    def __post_init__(self):
+        if not self.api_key:
+            raise ValueError("token is required")
+        if self.engine not in WRITESONIC_ENGINES:
+            raise ValueError("engine is invalid")
+        if self.language not in WRITESONIC_LANGUAGES:
+            raise ValueError("language is invalid")
+
+    def get_request_url(self):
+        return f'{self.base_url}{self.path}'
+
+
+@dataclass
+class ChatSonicPayload:
+    @dataclass
+    class Message:
+        is_sent: bool
+        message: str
+
+    input_text: str
+    history_data: List[Message] = field(default_factory=list)
+    enable_memory: bool = False
+    enable_google_results: bool = False
+
+    def push_response(self, message: str):
+        message_sent = self.Message(is_sent=True, message=self.input_text)
+        message_received = self.Message(is_sent=False, message=message)
+        self.history_data.append(message_sent)
+        self.history_data.append(message_received)
+        self.input_text = ''
+
+    def set_config(self, config: ChatSonicConfig):
+        self.enable_memory = config.enable_memory
+        self.enable_google_results = config.enable_google_results
+
+    def to_json(self):
+        return json.dumps(self.__dict__)
+
+
+@dataclass
+class ChatSonicResponse:
+    message: str
+    image_urls: List[str] = field(default_factory=list)
+
+    @staticmethod
+    def from_json(json_data):
+        return ChatSonicResponse(
+            message=json_data['message'],
+            image_urls=json_data['image_urls']
+        )
+
+
+class ChatSonic:
+    _config: ChatSonicConfig
+    _session = requests.Session()
+
+    def __init__(self, config: ChatSonicConfig):
+        self._config = config
+
+    def call(self, payload: ChatSonicPayload) -> (ChatSonicResponse, ChatSonicPayload):
+        url = self._config.get_request_url()
+
+        raw_response = self._session.post(
+            url=url,
+            headers=self.get_request_headers(),
+            params=self.get_request_params(),
+            data=payload.to_json()
+        )
+
+        raw_response.raise_for_status()
+
+        response = ChatSonicResponse.from_json(raw_response.json())
+        payload.push_response(response.message)
+
+        return response, payload
+
+    def chat(self, message: str) -> ChatSonicPayload:
+        payload = ChatSonicPayload(input_text=message);
+        payload.set_config(self._config)
+        return self.call(payload)
+
+    def get_request_headers(self):
+        return {'X-API-KEY': self._config.api_key}
+
+    def get_request_params(self):
+        return {
+            'engine': self._config.engine,
+            'language': self._config.language
+        }

edi_core/algo/silder_puzzle.py

@@ -0,0 +1,25 @@
+directions = [(0, 1), (1, 0), (0, -1), (-1, 0)]
+
+
+class SliderPuzzle:
+
+    def solve(self):
+        return
+
+
+class SearchNode:
+    board: list
+    free: tuple
+
+    step: 0
+
+    def get_distance(self):
+        distance = 0
+        for i in range(len(self.board)):
+            for j in range(len(self.board[i])):
+                if self.board[i][j] != i * len(self.board[i]) + j + 1:
+                    distance += abs(i - (self.board[i][j] - 1) // len(self.board[i])) + abs(
+                        j - (self.board[i][j] - 1) % len(self.board[i]))
+        return distance + self.step
+
+

edi_core/auto/control.py

@@ -0,0 +1,5 @@
+import pyautogui
+
+
+def write(text: str, interval: float = 0.1):
+    pyautogui.write(text, interval)

edi_core/db/tinydb.py

@@ -0,0 +1,23 @@
+from typing import List
+
+from tinydb import TinyDB
+from tinydb.table import Document
+
+
+class Tiny:
+    _db = None
+
+    def __init__(self, db_file):
+        self._db = TinyDB(db_file)
+
+    def insert(self, data: dict) -> int:
+        return self._db.insert(data)
+
+    def all(self) -> List[Document]:
+        return self._db.all()
+
+    def search(self, query) -> List[Document]:
+        return self._db.search(query)
+
+    def close(self):
+        self._db.close()

edi_core/file/file.py

@@ -0,0 +1,19 @@
+import os.path
+import shutil
+
+
+def isdir(path: str) -> bool:
+    return os.path.isdir(path)
+
+
+def isfile(path: str) -> bool:
+    return os.path.isfile(path)
+
+
+def move(src, dest):
+    shutil.move(src, dest)
+
+
+def delete_file(path: str):
+    if isfile(path):
+        os.remove(path)

edi_core/file/filetype.py

@@ -0,0 +1,5 @@
+import filetype
+
+
+def guess_mime(file):
+    return filetype.guess_mime(file)

edi_core/file/hash.py

@@ -0,0 +1,9 @@
+import hashlib
+
+algorithms_available = hashlib.algorithms_available
+
+
+def hash_str(data: str, algo='md5') -> str:
+    h = hashlib.new(algo)
+    h.update(data.encode('utf-8'))
+    return h.hexdigest()

edi_core/parser/toml_parser.py

@@ -0,0 +1,15 @@
+import toml
+
+
+def parse_toml_file(toml_file_path: str) -> dict:
+    with open(toml_file_path, 'r') as file:
+        return toml.load(file)
+
+
+def parse_toml_str(toml_str: str) -> dict:
+    return toml.loads(toml_str)
+
+
+def save_to_toml_file(path: str, data: dict):
+    with open(path, 'w') as file:
+        toml.dump(data, file)

edi_core/util/datetime.py

@@ -0,0 +1,6 @@
+import pendulum
+from pendulum import DateTime
+
+
+def now(tz='Asia/Shanghai') -> DateTime:
+    return pendulum.now(tz)

edi_core/util/environment.py

@@ -0,0 +1,13 @@
+import platform
+
+
+def get_platform():
+    return platform.platform()
+
+
+def get_system():
+    return platform.system()
+
+
+def is_windows():
+    return get_system() == 'Windows'

edi_core/util/pyproject.py

@@ -0,0 +1,13 @@
+from typing import Dict, Any
+
+from edi_core.parser.toml_parser import parse_toml_file
+
+
+class PyProject:
+    _root: str
+    _config: Dict[str, Any]
+    _config_file_name = "pyproject.toml"
+
+    def __init__(self, path: str):
+        self.path = path
+        self._config = parse_toml_file(f'{path}/{self._config_file_name}')

edi_core-0.3.0.dist-info/METADATA

@@ -0,0 +1,28 @@
+Metadata-Version: 2.1
+Name: edi-core
+Version: 0.3.0
+Summary: 
+License: MIT
+Author: LtttAZ
+Author-email: louis_zhang_1303@126.com
+Requires-Python: >=3.11,<4.0
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: dataclass-wizard (>=0.22.3,<0.23.0)
+Requires-Dist: filetype (>=1.2.0,<2.0.0)
+Requires-Dist: matplotlib (>=3.8.4,<4.0.0)
+Requires-Dist: pendulum (>=3.0.0,<4.0.0)
+Requires-Dist: pyautogui (>=0.9.54,<0.10.0)
+Requires-Dist: pytest (>=8.1.1,<9.0.0)
+Requires-Dist: requests (>=2.32.3,<3.0.0)
+Requires-Dist: requests-mock (>=1.12.1,<2.0.0)
+Requires-Dist: tinydb (>=4.8.0,<5.0.0)
+Requires-Dist: toml (>=0.10.2,<0.11.0)
+Requires-Dist: tomlkit (>=0.13.0,<0.14.0)
+Description-Content-Type: text/markdown
+
+# EDI
+
+

edi_core-0.3.0.dist-info/RECORD

@@ -0,0 +1,25 @@
+edi_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/ai/writesonic/__init__.py,sha256=QCibWjBh4QqvUeJUscjByC-zHd__21J6jfTPJcWJxO8,412
+edi_core/ai/writesonic/chatsonic.py,sha256=cmV1445v6cnvxY3kSkU5uLyLcPCHjWhvpEQ6zbfR9ug,3098
+edi_core/algo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/algo/silder_puzzle.py,sha256=lyPvnL32VAsTghrz2WBsPMPJqnMeFxbBgSt4khfPNVI,586
+edi_core/auto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/auto/control.py,sha256=W8JROSAAc900qUWNlCIQV9BikqeAtEycOXxA_7Eimqc,100
+edi_core/cloud/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/cloud/aliyun.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/db/tinydb.py,sha256=DB8trZcsxsLPSwlQjJemlOs68F4AcEMJJmyK0x7BXg0,463
+edi_core/file/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/file/file.py,sha256=VoydbmagVOSkE8EnJl8TpBckM7go_ayfjfz_pBVNfvU,282
+edi_core/file/filetype.py,sha256=yfGavZWzMkbmRUeYPFySZ7xVd1olbeCBc7oQVOMvtCE,77
+edi_core/file/hash.py,sha256=9YRLLDbPdTlZok6zGKfCLmDL9wXUpYVyHcTgz1Gt0Yc,200
+edi_core/parser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/parser/toml_parser.py,sha256=ll7RXZQsw68hx--PbtLR5kNh-52sOjq5m7x_XDmJAhI,328
+edi_core/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/util/datetime.py,sha256=sctJzZw_Uruq-qp6IW_JuqZvcJq_fXDbTMpfEwkAPlg,117
+edi_core/util/environment.py,sha256=rQmkKN5TWuhe-2NNGirQCeVW95dLz27pjeYDsB5yIII,175
+edi_core/util/pyproject.py,sha256=3mdC-a6rqx9zlZUHeGyDIjz0EcT_TmTINRKLxYRvIac,325
+edi_core-0.3.0.dist-info/METADATA,sha256=8vjVJGMonLuydRI4BB26XNrfgfVkDwRBDvPLnAQ4IH0,876
+edi_core-0.3.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+edi_core-0.3.0.dist-info/RECORD,,

edi_core-0.3.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 1.9.0
+Root-Is-Purelib: true
+Tag: py3-none-any