48 lines
2.3 KiB
Python
Executable File
48 lines
2.3 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
# Source: https://ewpratten.com/blog/better-dates-for-gitea-imports
|
|
|
|
import argparse
|
|
import sqlite3
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
if __name__ == "__main__":
|
|
ap = argparse.ArgumentParser()
|
|
ap.add_argument("database", help="Path to the Gitea SQLite database", type=Path)
|
|
ap.add_argument("repo_root", help="Path to Gitea's repository directory", type=Path)
|
|
ap.add_argument("--rewrite-archives", help="Update the archive timestamp to match the last commit", action="store_true")
|
|
args = ap.parse_args()
|
|
|
|
# Attempt to open the database
|
|
db_connection = sqlite3.connect(args.database)
|
|
db_cursor = db_connection.cursor()
|
|
|
|
# Process all know repositories
|
|
for repo in db_cursor.execute("SELECT owner_name, lower_name, archived_unix FROM repository").fetchall():
|
|
repo_path = args.repo_root / repo[0] / f"{repo[1]}.git"
|
|
|
|
if not repo_path.exists():
|
|
print(f"Skipping {repo_path} as it does not exist")
|
|
continue
|
|
|
|
# Obtain the first commit timestamp
|
|
# Some versions of git
|
|
first_commit = subprocess.run(["git", "-C", repo_path.as_posix(), "log", "--reverse", "--pretty=format:%ct"], capture_output=True, text=True).stdout.splitlines()[0]
|
|
first_commit_timestamp = int(first_commit)
|
|
|
|
# Obtain the last commit timestamp
|
|
last_commit = subprocess.run(["git", "-C", repo_path.as_posix(), "for-each-ref", "--sort=-authordate", "--count=1", "--format=%(authordate:unix)"], capture_output=True, text=True)
|
|
last_commit_timestamp = int(last_commit.stdout.strip())
|
|
|
|
# Update the database with the new timestamps
|
|
db_cursor.execute("UPDATE repository SET created_unix = ?, updated_unix = ? WHERE owner_name = ? AND lower_name = ?", (first_commit_timestamp, last_commit_timestamp, repo[0], repo[1]))
|
|
|
|
# If we are rewriting the archive timestamp, copy the "updated" timestamp to the "archived" timestamp
|
|
# Note: 0 means "not archived"
|
|
if args.rewrite_archives and repo[2] > 0:
|
|
db_cursor.execute("UPDATE repository SET archived_unix = ? WHERE owner_name = ? AND lower_name = ?", (last_commit_timestamp, repo[0], repo[1]))
|
|
|
|
# Commit the changes to the database
|
|
db_connection.commit()
|
|
db_connection.close()
|