mirror of
https://github.com/wcbing/wcbing-apt-repo.git
synced 2025-12-28 18:18:31 +08:00
81 lines
2.5 KiB
Python
Executable File
81 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
import subprocess
|
||
import os
|
||
import sqlite3
|
||
import logging
|
||
|
||
base_dir = "deb"
|
||
logging.basicConfig(
|
||
format="%(asctime)s %(message)s",
|
||
datefmt="%Y/%m/%d %H:%M:%S",
|
||
level=logging.INFO,
|
||
)
|
||
|
||
|
||
def download(url):
|
||
file_dir = os.path.join(base_dir, os.path.dirname(url))
|
||
if not os.path.exists(file_dir):
|
||
os.makedirs(file_dir)
|
||
file_path = os.path.join(base_dir, url.split("?")[0])
|
||
# 用 curl 模拟 apt 下载文件，User-Agent 来自 Debian 12
|
||
subprocess.run(
|
||
[
|
||
"curl",
|
||
"-H",
|
||
"User-Agent: Debian APT-HTTP/1.3 (2.6.1)",
|
||
"-fsLo",
|
||
file_path,
|
||
url,
|
||
]
|
||
)
|
||
|
||
|
||
def check_download(name, version, url, arch):
|
||
logging.info("%s:%s = %s", name, arch, version)
|
||
|
||
# connect to db
|
||
with sqlite3.connect(os.path.join("data", f"{base_dir}.db")) as conn:
|
||
cur = conn.cursor()
|
||
res = cur.execute(
|
||
f"SELECT version, url FROM {arch} WHERE name = ?", (name,)
|
||
).fetchall()
|
||
if len(res):
|
||
local_version = res[0][0]
|
||
local_url = res[0][1]
|
||
if local_version != version:
|
||
print(f"Update: {name}:{arch} ({local_version} -> {version})")
|
||
download(url, base_dir)
|
||
# wirte to db
|
||
cur.execute(
|
||
f"UPDATE {arch} SET version = ?, url = ? WHERE name = ?",
|
||
(version, url, name),
|
||
)
|
||
# remove old version
|
||
if local_url != url: # 针对固定下载链接
|
||
old_file_path = os.path.join(base_dir, local_url.split("?")[0])
|
||
if os.path.exists(old_file_path):
|
||
os.remove(old_file_path)
|
||
else:
|
||
print(f"AddNew: {name}:{arch} ({version})")
|
||
download(url, base_dir)
|
||
# wirte to db
|
||
cur.execute(
|
||
f"INSERT INTO {arch}(name, version, url) VALUES (?, ?, ?)",
|
||
(name, version, url),
|
||
)
|
||
conn.commit()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
args = os.sys.argv
|
||
if len(args) == 5:
|
||
check_download(args[1], args[2], args[3], args[4])
|
||
elif len(args) == 4:
|
||
check_download(args[1], args[2], args[3], "x86_64")
|
||
elif len(args) > 1:
|
||
logging.error(f"Unknown Args: {args[1:]}")
|
||
else:
|
||
print(f"Usage: {args[0]} <package_name> <version> <url> [arch]")
|
||
print("options:")
|
||
print(" arch: x86_64, arm64. default is x86_64")
|