-
Notifications
You must be signed in to change notification settings - Fork 5
/
main.py
39 lines (31 loc) · 1.29 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import sys
import os
from platform import python_version
# https://github.com/willmcgugan/rich
from rich.console import Console
from rich.theme import Theme
import pydbhub.dbhub as dbhub
if __name__ == '__main__':
custom_theme = Theme({
"info": "green",
"warning": "yellow",
"error": "bold red"
})
console = Console(theme=custom_theme)
if python_version()[0:3] < '3.7':
console.print("[ERROR] Make sure you have Python 3.7+ installed, quitting.\n\n", style="error")
sys.exit(1)
# Create a new DBHub.io API object
db = dbhub.Dbhub(config_file=f"{os.path.join(os.path.dirname(__file__), '..', 'config.ini')}")
# Retrieve the list of branches for a database
res, default_branch, err = db.Branches("justinclift", "Marine Litter Survey (Keep Northern Ireland Beautiful).sqlite")
if err is not None:
console.print(f"[ERROR] {err}", style="error")
sys.exit(1)
# Display the retrieved list of branches
default_branch_text = ""
console.print('Branches:', style="info")
for branche_name in res:
if branche_name == default_branch:
default_branch_text = ", default branch"
console.print(f" - {branche_name} (commits: {res[branche_name].commit_count}{default_branch_text})", style="info")