from peewee import *
import datetime

mysql_db = MySQLDatabase(
    "vps", user="dmy", password="abc123+_", host="zero.yoojoo.space", port=3306
)


def add_record(execution_datetime, server, speed):
    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):
    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
        )
    )