import argparse

import ntplib
import requests


# 使用前请先安装 ntplib 和 requests 两个依赖
def send_csv(username, filepath, disable_offset=False, is_dev=False):
    backend_prefix = "http://101.42.119.24:31415"
    if is_dev:
        backend_prefix = "http://localhost:31415"

    # 如果出现报错，尝试将这里的ntp_server替换为pool.ntp.org，或重试几次
    # 因为对时服务器不是足够稳定
    def get_offset(ntp_server="time.windows.com"):
        c = ntplib.NTPClient()
        response = c.request(ntp_server)
        local_offset = response.offset

        backend_res = requests.get(f"{backend_prefix}/time/offset")
        backend_offset = float(backend_res.content)
        # to ms
        return backend_offset - local_offset * 1000

    url = f"{backend_prefix}/users/{username}/sources/source_name/files"
    files = {
        "file": open(filepath, "rb")
    }
    if disable_offset:
        r = requests.post(url, files=files)
    else:
        offset = get_offset()
        r = requests.post(url, data={'offset': offset}, files=files)
    return r.content


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('-u', '--username', help='用户名', type=str, required=True)
    parser.add_argument('-f', '--file', help='日志文件路径', type=str, required=True)
    parser.add_argument('-O', '--disable-offset', help='不计算偏移量', action='store_true')
    parser.add_argument('-D', '--dev', help='开发模式（上传到本地数据库）', action='store_true')
    args = parser.parse_args()

    res = send_csv(args.username, args.file, args.disable_offset, args.dev)
    print(res)