edi-core 0.6.0__py3-none-any.whl → 0.8.0rc1__py3-none-any.whl

edi_core/cloud/README.md

@@ -0,0 +1 @@
+https://help.aliyun.com/zh/oss/developer-reference/get-started-with-oss-sdk-for-python-v2?spm=a2c4g.11186623.help-menu-31815.d_5_2_2_0_0.31152f8dIqy5FD

edi_core/cloud/oss.py

@@ -0,0 +1,64 @@
+from abc import ABC
+
+import alibabacloud_oss_v2 as oss
+from alibabacloud_oss_v2 import GetObjectMetaRequest, GetObjectMetaResult, GetObjectRequest
+from loguru import logger as log
+
+
+class OssClient(ABC):
+    def get_object_metadata(self, key: str):
+        pass
+
+    def get_object(self, key: str, target_path: str):
+        pass
+
+    def put_object(self, source_path: str, key: str):
+        pass
+
+    def delete_object(self, key: str):
+        pass
+
+
+class AliCloudOssClient(OssClient):
+    _config: oss.Config
+    _client: oss.Client
+    _bucket: str
+
+    def __init__(self, region: str, access_key: str, secret_key: str, endpoint: str, bucket: str):
+        config = oss.config.load_default()
+        config.region = region
+        config.credentials_provider = oss.credentials.StaticCredentialsProvider(access_key_id=access_key,
+                                                                                access_key_secret=secret_key)
+        config.endpoint = endpoint
+        self._config = config
+        self.print_config()
+        self._client = oss.Client(self._config)
+        self._bucket = bucket
+
+    def get_config(self) -> oss.Config:
+        return self._config
+
+    def get_object_metadata(self, key: str) -> GetObjectMetaResult:
+        request = GetObjectMetaRequest(bucket=self._bucket, key=key)
+        return self._client.get_object_meta(request)
+
+    def get_object(self, key: str, target_path: str):
+        log.debug(f"downloading {key} to {target_path}")
+        request = GetObjectRequest(bucket=self._bucket, key=key)
+        result = self._client.get_object(request)
+        with result.body as body_stream:
+            data = body_stream.read()
+            with open(target_path, 'wb') as f:
+                f.write(data)
+        log.debug(f"finished downloading {key} to {target_path}")
+
+    def put_object(self, source_path: str, key: str):
+        pass
+
+    def delete_object(self, key: str):
+        pass
+
+    def print_config(self):
+        config_attrs = vars(self._config)
+        config_str = "\n".join(f"{key}={value}" for key, value in config_attrs.items())
+        log.debug(f"OSS Config: {config_str}")

edi_core/file/audio.py

@@ -0,0 +1,36 @@
+import os
+import subprocess
+
+from edi_core.file.file import isdir, get_file_name_without_extension
+
+
+def extract_audio(file_path: str, output_path: str):
+    result = subprocess.run(
+        ['ffmpeg', '-i', file_path, '-q:a', '0', '-map', 'a', output_path],
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE
+    )
+    if result.returncode != 0:
+        raise RuntimeError(f"Failed to extract audio: {result.stderr.decode()}")
+
+
+def split_audio(source_file: str, dest_dir: str, seconds: int):
+    if not os.path.isfile(source_file):
+        raise ValueError(f"Source file does not exist: {source_file}")
+    if not isdir(dest_dir):
+        os.makedirs(dest_dir, exist_ok=True)
+
+    base_name = get_file_name_without_extension(source_file)
+    segment_pattern = os.path.join(dest_dir, f"{base_name}_%03d.mp3")
+
+    result = subprocess.run(
+        ['ffmpeg', '-i', source_file, '-f', 'segment', '-segment_time', str(seconds), '-c', 'copy', segment_pattern],
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE
+    )
+
+    if result.returncode != 0:
+        raise RuntimeError(f"Failed to split audio: {result.stderr.decode()}")
+
+    return sorted([os.path.join(dest_dir, f) for f in os.listdir(dest_dir) if f.startswith(base_name)],
+                  key=lambda x: x.lower())

edi_core/file/file.py

@@ -17,3 +17,11 @@ def move(src, dest):
 def delete_file(path: str):
     if isfile(path):
         os.remove(path)
+
+def get_file_name_without_extension(path: str) -> str:
+    base_name = os.path.basename(path)
+    name_without_extension, _ = os.path.splitext(base_name)
+    return name_without_extension
+
+def get_file_name_with_extension(path: str) -> str:
+    return os.path.basename(path)

edi_core/scm/git.py

@@ -0,0 +1,6 @@
+import git
+
+
+def clone_and_checkout(path, url, branch):
+    repo = git.Repo.clone_from(url=url, to_path=path)
+    repo.create_head(branch).checkout()

edi_core-0.8.0rc1.dist-info/METADATA

@@ -0,0 +1,27 @@
+Metadata-Version: 2.4
+Name: edi-core
+Version: 0.8.0rc1
+Summary: 
+License: MIT
+Author: LtttAZ
+Author-email: louis_zhang_1303@126.com
+Requires-Python: ==3.13.2
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Requires-Dist: alibabacloud-oss-v2 (>=1.2.0,<2.0.0)
+Requires-Dist: filetype (>=1.2.0,<2.0.0)
+Requires-Dist: gitpython (>=3.1.45,<4.0.0)
+Requires-Dist: loguru (>=0.7.3,<0.8.0)
+Requires-Dist: matplotlib (>=3.10.7,<4.0.0)
+Requires-Dist: packaging (>=25.0,<26.0)
+Requires-Dist: pendulum (>=3.1.0,<4.0.0)
+Requires-Dist: pyautogui (>=0.9.54,<0.10.0)
+Requires-Dist: pydantic (>=2.12.3,<3.0.0)
+Requires-Dist: python-dotenv (>=1.2.1,<2.0.0)
+Requires-Dist: requests (>=2.32.5,<3.0.0)
+Requires-Dist: tinydb (>=4.8.2,<5.0.0)
+Description-Content-Type: text/markdown
+
+# EDI
+
+[PYPI](https://pypi.org/project/edi-core/)

edi_core-0.8.0rc1.dist-info/RECORD

@@ -0,0 +1,21 @@
+edi_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/auto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/auto/control.py,sha256=W8JROSAAc900qUWNlCIQV9BikqeAtEycOXxA_7Eimqc,100
+edi_core/cloud/README.md,sha256=eORYs2CT7hkm2i2A7xmREaW4I_Scl1uTbYNwu8TzhhI,151
+edi_core/cloud/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/cloud/oss.py,sha256=8w5CB1INo1e69VJSjPhbvnqKOUig3JjF3fDMcags1uE,2151
+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/audio.py,sha256=R9SzLTJ6_nTjs5wsmziOo76sij0ZZLEXKaiJdeFjDzA,1298
+edi_core/file/file.py,sha256=JE69sgThTx-bzSdmn-dnkSIGX-58hRXf3w7PbqzDLMs,558
+edi_core/file/filetype.py,sha256=yfGavZWzMkbmRUeYPFySZ7xVd1olbeCBc7oQVOMvtCE,77
+edi_core/file/hash.py,sha256=9YRLLDbPdTlZok6zGKfCLmDL9wXUpYVyHcTgz1Gt0Yc,200
+edi_core/scm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edi_core/scm/git.py,sha256=-TVM0_IDd34zAuJPZweksTItzqhUV75DtG4YXe9qQvM,150
+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-0.8.0rc1.dist-info/METADATA,sha256=4Q4ExOejyDWh9PQrmuNmepq42flB08ThmbMLGmmx2LA,861
+edi_core-0.8.0rc1.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+edi_core-0.8.0rc1.dist-info/RECORD,,

{edi_core-0.6.0.dist-info → edi_core-0.8.0rc1.dist-info}/WHEEL

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

edi_core/ai/writesonic/__init__.py

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

edi_core/ai/writesonic/chatsonic.py

@@ -1,110 +0,0 @@
-import dataclasses
-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 ChatSonicRequest:
-    @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_dict(self):
-        return dataclasses.asdict(self)
-
-
-@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: ChatSonicRequest) -> (ChatSonicResponse, ChatSonicRequest):
-        url = self._config.get_request_url()
-        payload.set_config(self._config)
-
-        raw_response = self._session.post(
-            url=url,
-            headers=self.get_request_headers(),
-            params=self.get_request_params(),
-            json=payload.to_dict(),
-            timeout=60
-        )
-
-        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, history_data: List[ChatSonicRequest.Message] = []) -> (
-            ChatSonicResponse, ChatSonicRequest):
-        payload = ChatSonicRequest(input_text=message, history_data=history_data)
-        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

@@ -1,25 +0,0 @@
-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/parser/toml_parser.py

@@ -1,15 +0,0 @@
-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/pyproject.py

@@ -1,13 +0,0 @@
-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.6.0.dist-info/METADATA

@@ -1,28 +0,0 @@
-Metadata-Version: 2.1
-Name: edi-core
-Version: 0.6.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
-
-[PYPI](https://pypi.org/project/edi-core/)

edi_core-0.6.0.dist-info/RECORD

@@ -1,25 +0,0 @@
-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=PzGYZQgjQD8wnnopJc-G455cvqxjiSognfoOfuWVY9Y,412
-edi_core/ai/writesonic/chatsonic.py,sha256=RHzmE2W_Oiomcn0YEo0v9t-F5SVQ-9bEOtk5tykBFHA,3239
-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.6.0.dist-info/METADATA,sha256=bN6WFxSwQtNGR4Ij2HXDOzy47hbcF1QG4jXoB5nCkso,918
-edi_core-0.6.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-edi_core-0.6.0.dist-info/RECORD,,