crxsnake 0.0.0__py3-none-any.whl

crxsnake/__init__.py

@@ -0,0 +1,19 @@
+from .logger import logger
+from .cogs import load_cogs
+from .files import read_json, write_json
+from .tortoise import Database
+from .misc import *
+
+from ...issues.hotline import IssueHotline
+from ...issues.crx import IssueCRX
+
+
+__all__ = [
+    IssueHotline,
+    IssueCRX,
+    logger,
+    Database,
+    read_json,
+    write_json,
+    load_cogs,
+]

crxsnake/cogs.py

@@ -0,0 +1,16 @@
+from sys import path
+from pathlib import Path
+from rich.console import Console
+
+
+async def load_cogs(bot, directory="src"):
+    directory_path = Path(directory)
+    path.append(str(directory_path))
+
+    for entry in directory_path.iterdir():
+        for filename in entry.iterdir():
+            if filename.suffix == ".py" and entry.name != "utils":
+                try:
+                    bot.load_extension(f"{entry.name}.{filename.stem}")
+                except Exception:
+                    Console().print_exception(show_locals=True)

crxsnake/files.py

@@ -0,0 +1,37 @@
+from typing import Optional, Any
+from json import loads, dumps
+
+from aiofiles import open
+from rich.console import Console
+
+
+async def read_json(path: str, *keys: str) -> Optional[Any]:
+    """
+    Read data from json file.
+    """
+    try:
+        async with open(path, "r", encoding="utf-8") as file:
+            file_content = await file.read()
+            file_data = loads(file_content)
+
+            for key in keys:
+                if key not in file_data:
+                    return None
+
+                file_data = file_data[key]
+            return file_data
+
+    except Exception:
+        Console().print_exception(show_locals=True)
+
+
+async def write_json(path: str, data: Any) -> None:
+    """
+    Write data to json file.
+    """
+    try:
+        async with open(path, "w", encoding="utf-8") as file:
+            await file.write(dumps(data, ensure_ascii=False, indent=2))
+
+    except Exception:
+        Console().print_exception(show_locals=True)

crxsnake/issues/crx.py

@@ -0,0 +1,14 @@
+from typing import List, Dict, Any
+
+
+class IssueCRX:
+    async def create_items(
+        self, issue_type: str, items_list: List[Dict[str, Any]]
+    ) -> Dict[str, List[Dict[str, Any]]]:
+        """
+        Create a dictionary with generated codes and processed items.
+        """
+
+        issue_dict = {"items": [], "cars": [], "sets": []}
+        issue_dict[issue_type].extend(items_list)
+        return issue_dict

crxsnake/issues/hotline.py

@@ -0,0 +1,77 @@
+from random import choices
+from string import ascii_uppercase, ascii_letters, digits
+from typing import List, Dict, Any, AsyncGenerator
+
+
+class IssueHotline:
+    MAX_STACK = 100
+
+    async def __generate_code(self) -> str:
+        part1 = "".join(choices(ascii_uppercase, k=2))
+        part2 = "".join(choices(ascii_letters + digits, k=4))
+        part3 = "".join(choices(ascii_letters + digits, k=4))
+        return f"{part1}-{part2}-{part3}"
+
+    async def __generate_count(
+        self, items_list: List[Dict[str, Any]]
+    ) -> AsyncGenerator[Dict[str, Any], None]:
+        """
+        Generate item counts based on the MAX_STACK value.
+        Splits items into full stacks and the remainder.
+        """
+
+        for item in items_list:
+            item_count = item.get("m_count", 0)
+            item_name = item.get("m_item", "")
+            full_stacks, remainder = divmod(item_count, self.MAX_STACK)
+
+            for _ in range(full_stacks):
+                yield {"m_item": item_name, "m_count": self.MAX_STACK}
+
+            if remainder > 0:
+                yield {"m_item": item_name, "m_count": remainder}
+
+    async def create_items(
+        self, issue_name: str, items_list: List[Dict[str, Any]]
+    ) -> Dict[str, Any]:
+        """
+        Create a dictionary with generated codes and processed items.
+        """
+
+        processed_items = [item async for item in self.__generate_count(items_list)]
+        return {
+            "m_CodeArray": [
+                {
+                    "m_code": await self.__generate_code(),
+                    "m_name": issue_name,
+                    "m_type": "item",
+                    "m_itemsArray": processed_items,
+                    "m_vehicles": {},
+                    "m_teleport_position": "",
+                    "m_give_zone_positions": [],
+                    "m_give_zone_positions_forbidden": [],
+                }
+            ]
+        }
+
+    async def create_vehicle(
+        self, issue_name: str, items_dict: Dict[str, Any]
+    ) -> Dict[str, Any]:
+        """
+        Create a dictionary with generated code and vehicle data.
+        """
+
+        return {
+            "m_CodeArray": [
+                {
+                    "m_code": await self.__generate_code(),
+                    "m_name": issue_name,
+                    "m_type": "vehicle",
+                    "m_itemsArray": [],
+                    "m_vehicles": items_dict,
+                    "m_teleport_position": "",
+                    "m_give_zone_positions": [],
+                    "m_give_zone_positions_forbidden": [],
+                }
+            ]
+        }

crxsnake/logger.py

@@ -0,0 +1,28 @@
+import sys
+from loguru import logger
+
+
+logger.remove(0)
+logger.add(
+    sys.stderr,
+    format="<blue>[{time:HH:mm:ss}]</blue> <yellow>({level})</yellow> <red>|→</red> <green>{message}</green>",
+    colorize=True,
+    level="INFO",
+    filter=lambda record: record["level"].name == "INFO",
+)
+
+logger.add(
+    sys.stderr,
+    format="<blue>[{time:HH:mm:ss}]</blue> <yellow>({level})</yellow> |> <yellow>{message}</yellow>",
+    colorize=True,
+    level="WARNING",
+    filter=lambda record: record["level"].name == "WARNING",
+)
+
+logger.add(
+    sys.stderr,
+    format="<blue>[{time:HH:mm:ss}]</blue> <red>({level})|→ {message}</red>",
+    colorize=True,
+    level="ERROR",
+    filter=lambda record: record["level"].name == "ERROR",
+)

crxsnake/misc.py

@@ -0,0 +1,16 @@
+from datetime import datetime
+
+red = 16711680
+green = 65280
+blue = 255
+yellow = 16776960
+purple = 9109759
+trans = 2829617
+
+timestamp = int(datetime.now().timestamp())
+time_short = f"<t:{timestamp}:d>"
+time_long_date = f"<t:{timestamp}:D>"
+time_short_time = f"<t:{timestamp}:t>"
+time_long_time = f"<t:{timestamp}:T>"
+time_full = f"<t:{timestamp}:f>"
+time_relative = f"<t:{timestamp}:R>"

crxsnake/tortoise.py

@@ -0,0 +1,28 @@
+from tortoise import Tortoise
+from rich.console import Console
+
+
+class Database:
+    DB_URL = "sqlite://settings/database/database.db"
+    DB_MODEL = {"models": ["src.utils"]}
+
+    async def db_connect(self) -> None:
+        """
+        Connect to the database.
+        """
+        try:
+            await Tortoise.init(db_url=self.DB_URL, modules=self.DB_MODEL)
+            await Tortoise.generate_schemas()
+
+        except Exception:
+            Console().print_exception(show_locals=True)
+
+    async def db_disconnect(self) -> None:
+        """
+        Disconnect from the database.
+        """
+        try:
+            await Tortoise.close_connections()
+
+        except Exception:
+            Console().print_exception(show_locals=True)

crxsnake-0.0.0.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 CRX-DEV
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

crxsnake-0.0.0.dist-info/METADATA

@@ -0,0 +1,5 @@
+Metadata-Version: 2.1
+Name: crxsnake
+Version: 0.0.0
+License-File: LICENSE
+

crxsnake-0.0.0.dist-info/RECORD

@@ -0,0 +1,14 @@
+crxsnake/__init__.py,sha256=F8W6qG38P8nF8nfCCPfEdPpzJdFX2VyQdzVUvAsPXaM,365
+crxsnake/cogs.py,sha256=gXC81xWcpUZ-YOpI8BqiUoy7s8AInLr0M1asrSFAxEg,557
+crxsnake/files.py,sha256=M6uCktEFaDUuVH3CHUcr9bITZa_ruEsAjIRqf_5gM3o,997
+crxsnake/logger.py,sha256=-TXqHfagCV-TrnwJrucKEa61BRePW34sep0K9RCbHn8,764
+crxsnake/misc.py,sha256=VAv6FTQ-Ekwudtm8iLKjSu3WFSJuTw8Nrs_Gclo1W_c,402
+crxsnake/tortoise.py,sha256=R0IwdfvsS5FGUXlGVIdI1I48_Xde_qz_Rd8E2e_NNHY,768
+crxsnake/issues/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+crxsnake/issues/crx.py,sha256=mhYX4iWmyPw5rV60rb0m18HAy7s72aTFwA-MIsdXV1E,431
+crxsnake/issues/hotline.py,sha256=tLUVp59O8YesXxlmVxOeLcIjDN7RiwWTZGCVKIxXR-Q,2683
+crxsnake-0.0.0.dist-info/LICENSE,sha256=xt4Ru6tOCU8e2wVlx_lgx7wh-sNLKbiCbmANzr2e3cc,1085
+crxsnake-0.0.0.dist-info/METADATA,sha256=dyV-kv1iq56dUJAdhzbldudAZP3ajj8n9IIte7eHKf4,80
+crxsnake-0.0.0.dist-info/WHEEL,sha256=HiCZjzuy6Dw0hdX5R3LCFPDmFS4BWl8H-8W39XfmgX4,91
+crxsnake-0.0.0.dist-info/top_level.txt,sha256=GOgG6tMH05czsfM6y-Tvm9LUSd-0PPuuuYkkkbWHZjQ,9
+crxsnake-0.0.0.dist-info/RECORD,,

crxsnake-0.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (72.2.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

crxsnake-0.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+crxsnake