download_speed/download_speed/model.py

37 lines
1.1 KiB
Python

from peewee import *
import datetime
from download_speed import config
config.init_config()
mysql_addr = config.get('mysql_addr')
mysql_db = MySQLDatabase(
"vps", user="dmy", password="abc123+_", host=mysql_addr, port=3306
)
def add_record(execution_datetime, server, speed):
if not config.get('mysql_enable'):
return
execution_datetime_utc = execution_datetime - datetime.timedelta(
hours=8
) # 转换为UTC时间。
# print(now_timestamp)
mysql_db.execute_sql(
'insert into vps_download_speed(server,speed,datetime) values("{server}",{speed},"{datetime}")'.format(
server=server, speed=speed, datetime=execution_datetime_utc
)
)
def add_ping_record(execution_datetime, server, ping):
if not config.get('mysql_enable'):
return
execution_datetime_utc = execution_datetime - datetime.timedelta(
hours=8
) # 转换为UTC时间。
mysql_db.execute_sql(
'insert into vps_ping(server,ping,datetime) values("{server}",{ping},"{datetime}")'.format(
server=server, ping=ping, datetime=execution_datetime_utc
)
)