mirror of
https://github.com/Buriburizaem0n/nezha_domains.git
synced 2026-05-06 05:38:50 +00:00
naiba
124 lines
3.7 KiB
Python
124 lines
3.7 KiB
Python
import os
|
|
import time
|
|
import requests
|
|
from github import Github
|
|
|
|
ATOMGIT_API = "https://api.atomgit.com/api/v5"
|
|
ATOMGIT_OWNER = "naiba"
|
|
ATOMGIT_REPO = "nezha-dashboard"
|
|
GITHUB_REPO = "nezhahq/nezha"
|
|
|
|
|
|
def get_github_latest_release():
|
|
g = Github()
|
|
repo = g.get_repo(GITHUB_REPO)
|
|
release = repo.get_latest_release()
|
|
if not release:
|
|
print("No releases found.")
|
|
return
|
|
|
|
print(f"Latest release tag is: {release.tag_name}")
|
|
print(f"Latest release info is: {release.body}")
|
|
files = []
|
|
for asset in release.get_assets():
|
|
url = asset.browser_download_url
|
|
name = asset.name
|
|
|
|
response = requests.get(url)
|
|
if response.status_code == 200:
|
|
with open(name, "wb") as f:
|
|
f.write(response.content)
|
|
print(f"Downloaded {name}")
|
|
else:
|
|
print(f"Failed to download {name}")
|
|
files.append(get_abs_path(name))
|
|
sync_to_atomgit(release.tag_name, release.body, files)
|
|
|
|
|
|
def sync_to_atomgit(tag, body, files):
|
|
access_token = os.environ["ATOMGIT_PAT"]
|
|
release_api_uri = f"{ATOMGIT_API}/repos/{ATOMGIT_OWNER}/{ATOMGIT_REPO}/releases"
|
|
|
|
release_data = {
|
|
"access_token": access_token,
|
|
"tag_name": tag,
|
|
"name": tag,
|
|
"body": body,
|
|
"prerelease": False,
|
|
"target_commitish": "master",
|
|
}
|
|
|
|
release_resp = None
|
|
for attempt in range(3):
|
|
try:
|
|
release_resp = requests.post(release_api_uri, json=release_data, timeout=30)
|
|
release_resp.raise_for_status()
|
|
break
|
|
except requests.exceptions.Timeout:
|
|
print(
|
|
f"Create release timed out, retrying in 30s... (attempt {attempt + 1})"
|
|
)
|
|
time.sleep(30)
|
|
except requests.exceptions.RequestException as err:
|
|
print(f"Create release failed: {err}")
|
|
if release_resp is not None:
|
|
print(f"Response: {release_resp.text}")
|
|
break
|
|
|
|
if release_resp is None or release_resp.status_code not in (200, 201):
|
|
print("Failed to create release on AtomGit, aborting.")
|
|
return
|
|
|
|
print(f"Created release {tag} on AtomGit")
|
|
|
|
for file_path in files:
|
|
upload_asset(access_token, tag, file_path)
|
|
|
|
print("Sync is completed!")
|
|
|
|
|
|
def upload_asset(access_token, tag, file_path):
|
|
file_name = os.path.basename(file_path)
|
|
upload_url_api = (
|
|
f"{ATOMGIT_API}/repos/{ATOMGIT_OWNER}/{ATOMGIT_REPO}"
|
|
f"/releases/{tag}/upload_url?file_name={file_name}"
|
|
)
|
|
|
|
for attempt in range(3):
|
|
try:
|
|
resp = requests.get(
|
|
upload_url_api,
|
|
headers={"Authorization": f"Bearer {access_token}"},
|
|
timeout=30,
|
|
)
|
|
resp.raise_for_status()
|
|
upload_info = resp.json()
|
|
|
|
obs_url = upload_info["url"]
|
|
obs_headers = upload_info["headers"]
|
|
|
|
with open(file_path, "rb") as f:
|
|
put_resp = requests.put(
|
|
obs_url, headers=obs_headers, data=f, timeout=120
|
|
)
|
|
|
|
if put_resp.text.strip() == "success" or put_resp.status_code in (200, 201):
|
|
print(f"Uploaded {file_name}")
|
|
return
|
|
else:
|
|
print(
|
|
f"Upload {file_name} failed: {put_resp.status_code} {put_resp.text}"
|
|
)
|
|
except requests.exceptions.RequestException as err:
|
|
print(f"Upload {file_name} attempt {attempt + 1} failed: {err}")
|
|
time.sleep(10)
|
|
|
|
print(f"Failed to upload {file_name} after 3 attempts")
|
|
|
|
|
|
def get_abs_path(path):
|
|
return os.path.join(os.getcwd(), path)
|
|
|
|
|
|
get_github_latest_release()
|