#!/usr/bin/python3

import sys
import psycopg2

# define any local helper functions here

### set up some globals

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

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

# process command-line args

q = '''
select p.code, t.code
from Programs p
join Program_enrolments e on e.program = p.id
join Terms t on e.term = t.id
where e.student = %s
order by t.starting
'''

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

  # manipulate database, e.g.
  cur = db.cursor();
  cur.execute(q, [zid])
  for res in cur.fetchall():
    print(res[0],res[1])

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