update: 优化代码

This commit is contained in:
MarkusJoe
2022-09-14 12:46:27 +08:00
parent 2822b61d8e
commit 1824ecb030
1 file changed
+6 -12
+6 -12
View File
@@ -12,20 +12,14 @@ import os
class Config:
"""
database type. Default is sqlite
sqlite file path is "./src/db/data.sqlite", it is not exists, before you run server first
database file path is "./src/db/data.sqlite", it is not exists, before you run server first
remote sqlite file url is "https://pac.rtst.tech/static_file_hosting/static/counter/data.sqlite"
you can download it manually
if you want to use mysql then replace "mysql-[Username]:[password]@[host]:[port]/[database]" with "sqlite"
if you want to use mysql then replace "mysql|[Username]:[password]@[host]:[port]/[database]" with "sqlite"
example: mysql-root:114514@127.0.0.1:3306/homo
port is required
"""
database = "sqlite"
if database != 'sqlite':
os.environ['m_user'] = database.split('-')[-1].split('@')[0].split(':')[0]
os.environ['m_password'] = database.split('-')[-1].split('@')[0].split(':')[1]
os.environ['m_host'] = database.split('-')[-1].split('@')[-1].split('/')[0].split(':')[0]
os.environ['m_port'] = database.split('-')[-1].split('@')[-1].split('/')[0].split(':')[1]
os.environ['m_database'] = database.split('-')[-1].split('@')[-1].split('/')[-1]
if 'sqlite' in database:
os.environ['counter_db_type'] = 'sqlite'
elif 'mysql' in database:
os.environ['counter_db_type'] = 'mysql'