Exemplo n.º 1
0
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.")
Exemplo n.º 2
0
 def test_class_factory(self):
     for item in self.class_list:
         self.assertIsNotNone(class_factory(item))