script to log the status of phabricator to a DB

scripts/metrics/phab_monitoring.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python3
+
+import psycopg2
+from phabricator import Phabricator
+import os
+from typing import Optional
+import datetime
+
+
+def phab_up() -> Optional[Phabricator]:
+    """Try to connect to phabricator to see if the server is up.
+
+    Returns None if server is down.
+    """
+    print("Checking Phabricator status...")
+    try:
+        phab = Phabricator()
+        phab.update_interfaces()
+        print("Phabricator is up...")
+        return phab
+    except exception:
+        pass
+        print("Phabricator is down...")
+    return None
+
+
+def log_phab_status(up: bool, conn: psycopg2.extensions.connection):
+    """log the phabricator status to the database."""
+    print("Writing Phabricator status to database...")
+
+    cur = conn.cursor()
+    cur.execute(
+        "INSERT INTO phab_status (timestamp, status) VALUES (%s,%s);",
+        (datetime.datetime.now(), up),
+    )
+    conn.commit()
+
+
+def connect_to_db() -> psycopg2.extensions.connection:
+    """Connect to the database, create tables as needed."""
+    conn = psycopg2.connect(
+        "host=127.0.0.1 sslmode=disable dbname=stats user={} password={}".format(
+            os.environ["PGUSER"], os.environ["PGPASSWORD"]
+        )
+    )
+    cur = conn.cursor()
+    cur.execute(
+        "CREATE TABLE IF NOT EXISTS phab_status (timestamp timestamp, status boolean);"
+    )
+    conn.commit()
+    return conn
+
+
+def phab_monitoring():
+    """Main function of monitoring the phabricator server."""
+    conn = connect_to_db()
+    phab = phab_up()
+    log_phab_status(phab is not None, conn)
+    print("Completed, exiting...")
+
+
+if __name__ == "__main__":
+    phab_monitoring()

scripts/requirements.txt

@@ -6,4 +6,5 @@ phabricator==0.8.1
 pyaml==20.4.0
 requests==2.25.1
 unidiff==0.6.0
-python-benedict==0.23.2
+python-benedict==0.23.2
+psycopg2