def main():
jobs = argparse.ArgumentParser(description=DESCRIPTION, epilog=EPILOG)
#jobs.add_argument("-v", action="store_true", help="verbose help")
group = jobs.add_mutually_exclusive_group()
group.add_argument("--add", choices=CHOICES, help=ADD_HELP)
group.add_argument("--show", choices=CHOICES_PLURAL, help=SHOW_HELP)
group.add_argument("--update", choices=CHOICES, help=UPDATE_HELP)
group.add_argument("--delete", choices=CHOICES, help=DELETE_HELP)
cli_args = jobs.parse_args()
if cli_args.add:
class_object = class_factory(cli_args.add)
# Need to figure this out...
update_class(class_object, class_object.properties)
elif cli_args.show:
objects_to_show = show(cli_args.show)
if objects_to_show:
print_to_screen(objects_to_show)
else:
print("No {} in the database. Try adding one.".format(cli_args.show))
elif cli_args.update:
class_object = class_factory(cli_args.update)
all_classes = get_all_objects_in_db(class_object)
wanted_class_object = selection_screen(all_classes)
update_class(wanted_class_object, wanted_class_object.properties)
elif cli_args.delete:
class_object = class_factory(cli_args.delete)
all_classes = get_all_objects_in_db(class_object)
wanted_class_object = selection_screen(all_classes)
delete_class_object(wanted_class_object)
else:
objects_to_show = show(JOB_POSTINGS)
if objects_to_show:
print_to_screen(objects_to_show)
else:
print("No jobpostings in the database. Try adding one.")
def test_get_all_objects_in_db(self):
for item in self.classes:
self.assertTrue(isinstance(get_all_objects_in_db(item()), list))
