diff --git a/src/db/__init__.py b/src/db/__init__.py index 813821d..4238412 100644 --- a/src/db/__init__.py +++ b/src/db/__init__.py @@ -11,30 +11,6 @@ from src.config import Config database = Config.database - -def split_(origin: str, type_: int = 0) -> dict: - """ - - :param origin: - :param type_: 0 -> Mysql - :return: - """ - origin = origin.split('://')[-1] - host = origin.split('@')[-1].split('/')[0].split(':')[0] - port = origin.split('@')[-1].split('/')[0].split(':')[1] - if type_ == 0: # Mysql - database_ = origin.split('@')[-1].split('/')[-1] - user = origin.split('@')[0].split(':')[0] - password = origin.split('@')[0].split(':')[1] - return { - 'host': host, - 'port': port, - 'database': database_, - 'user': user, - 'password': password - } - - if database == 'sqlite': import sqlite3 as operator from src.utils.t_download import download @@ -42,12 +18,7 @@ if database == 'sqlite': if not os.path.exists('./src/db/data.sqlite'): download('https://markusjoe.github.io/static_file_hosting/static/counter/data.sqlite') from src.db.db import SQLite as Database -else: - import pymysql as operator - - result = split_(database) - for k, v in zip(result.keys(), result.values()): - os.environ[f"m_{k}"] = v - from src.db.db import MySQL as Database - +elif 'redis' in database: + import redis as operator + from src.db.db import Redis as Database __all__ = [Database, operator]