#!/usr/bin/python3

import sys
import psycopg2

# define any local helper functions here

### set up some globals

usage = f"Usage: {sys.argv[0]} Term ..."
db = None

if len(sys.argv) < 2:
	print(usage)
	exit()
term = sys.argv[1]

# process command-line args

try:
  # connect to database, e.g.
  db = psycopg2.connect("dbname=mymy")

  # manipulate database, e.g.
  cur = db.cursor();
  cur.execute("select id from Terms where code = %s",[term])
  t = cur.fetchone()
  if not t:
    print("No such term")
    exit()

  cur.execute("select count(*) from Courses where term=%s",[t[0]]);
  res = cur.fetchone()
  if not res:
    print("No courses?")
  else:
    print(f"{res[0]} courses in {term}")

except Exception as err:
  print("DB error: ", err)
finally:
  if db:
    db.close()