diff --git a/bin/test-school-code b/bin/test-school-code new file mode 100755 index 0000000000000000000000000000000000000000..52c0591561eef6b1e4233ce6452d70835e885177 --- /dev/null +++ b/bin/test-school-code @@ -0,0 +1,25 @@ +#!/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()