Testy: Nastavování kódů škol

bin/test-school-code

+#!/usr/bin/env python3
+
+import argparse
+
+import mo.db as db
+import mo.util
+from mo.util import die
+
+parser = argparse.ArgumentParser(description='Přidělí kód škole')
+parser.add_argument('--red-izo', type=str, required=True, help='RED_IZO školy')
+parser.add_argument('--code', type=str, required=True, help='kód k přidělení')
+
+args = parser.parse_args()
+
+mo.util.init_standalone()
+sess = db.get_session()
+
+schools = sess.query(db.School).filter_by(red_izo=args.red_izo).all()
+if not schools:
+    die("Zadané RED_IZO nemá žádná škola")
+if len(schools) > 1:
+    die("Zadané RED_IZO má více škol")
+
+schools[0].place.code = args.code
+sess.commit()