Adds list_repos and list_commits functions to data.py (#16)

Author: Zoom30

icds/data.py

@@ -10,6 +10,14 @@ def get_repo_by_name(db, name: str) -> DbRepo | None:
     return repo
 
 
+def list_repos(db, limit=None) -> list[DbRepo] | None:
+    statement = select(DbRepo)
+    if limit:
+        statement = statement.limit(limit)
+    repos = db.exec(statement).all()
+    return repos
+
+
 def get_or_create_db_repo(db, repo, repo_path) -> DbRepo:
     remote_url = repo.remotes.origin.url
     repo_name = remote_url.split("/")[-1].split(".")[0]
@@ -30,6 +38,14 @@ def get_repo_name_by_id(db, repo_id) -> str:
     return db.exec(select(DbRepo).where(DbRepo.id == repo_id)).first().name
 
 
+def list_commits(db, repo_id, limit=None) -> list[RepoCommit] | None:
+    statement = select(RepoCommit).where(RepoCommit.repo_id == repo_id)
+    if limit:
+        statement = statement.limit(limit)
+    commits = db.exec(statement).all()
+    return commits
+
+
 def search_commits(
     db, query, repo_name, file, author, start_date, end_date
 ) -> list[RepoCommit]: