def help(self):
entities_available = ""
em = EntityManager()
for name in em.get_all_entities():
entities_available += "\t - " + name + "\n"
command_executable = "{} {}".format(sys.executable, sys.argv[0])
usage = "{} update [entity name] [id][".format(command_executable)
print("""
Update an existing entity (location, user, etc...)
You will be prompted to complete the necessary fields, once you selected an entity
Usage: {}
Available entities:
{}
Exmaple usage:
""".format(usage, entities_available))
class Update(Command):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.entity_manager = EntityManager()
def help(self):
entities_available = ""
em = EntityManager()
for name in em.get_all_entities():
entities_available += "\t - " + name + "\n"
command_executable = "{} {}".format(sys.executable, sys.argv[0])
usage = "{} update [entity name] [id][".format(command_executable)
print("""
Update an existing entity (location, user, etc...)
You will be prompted to complete the necessary fields, once you selected an entity
Usage: {}
Available entities:
{}
Exmaple usage:
""".format(usage, entities_available))
def run(self, *args):
if len(args) < 1:
return self.help()
entity_name = args[0]
entity = self.entity_manager.get_entity(entity_name)
logging.debug("EM: %s", self.entity_manager)
logging.debug("Available entities: %s",
self.entity_manager.get_all_entities())
if not entity:
print("Couldn't find the class you are looking for")
return self.help()
if len(args) != 2:
print("You must only provide a single id to update")
return
entity = entity.find(args[1])
if entity is None:
print("Couldn't find {} {}".format(args[0], args[2]))
logging.debug("UPDATING ENTITY: %s", entity)
print("Updating {}:\n\n".format(entity_name))
self.ask_for_fields(entity)
for name, relationship in entity.relationships.items():
if isinstance(relationship, ManyToMany):
pass
# Do something for many to many
elif isinstance(relationship, OneToMany):
#get all remote entities
available_data = list(relationship.find_all())
available_data = list(
map(lambda x: (x.render_excerpt(), x), available_data))
logging.debug(available_data)
questions = [
inquirer.Checkbox(
"relation",
message="Which {}?".format(name),
choices=available_data,
)
]
res = inquirer.prompt(questions)
logging.debug(res)
setattr(entity, name, res["relation"])
elif isinstance(relationship, OneToOne):
#get all remote entities
available_data = list(relationship.find_all())
available_data = list(
map(lambda x: (x.render_excerpt(), x), available_data))
logging.debug(available_data)
questions = [
inquirer.List(
"relation",
message="Which {}?".format(name),
choices=available_data,
)
]
res = inquirer.prompt(questions)
logging.debug(res)
setattr(entity, name, res["relation"])
#TODO add resume plus confirm
entity.save()
print("{} {} updated succesfully".format(entity_name, entity.id))
def ask_for_fields(self, entity):
fields = entity.fields
#filter out all relationship fields to not ask for them
fields = {k: v for (k, v) in fields.items() if v is not "relationship"}
questions = map(lambda f: inquirer.Text(f, message="{}".format(f)),
fields)
answers = inquirer.prompt(questions)
logging.debug("Creating entity (%s) with values %s", entity, answers)
for field, value in answers.items():
setattr(entity, field, value)
class Destroy(Command):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.entity_manager = EntityManager()
def help(self):
entities_available = ""
em = EntityManager()
for name in em.get_all_entities():
entities_available += "\t - " + name + "\n"
command_executable = "{} {}".format(sys.executable, sys.argv[0])
usage = "{} update [entity name] [id][".format(command_executable)
print("""
Update an existing entity (location, user, etc...)
You will be prompted to complete the necessary fields, once you selected an entity
Usage: {}
Available entities:
{}
Exmaple usage:
""".format(usage, entities_available))
def run(self, *args):
if len(args) < 1:
return self.help()
entity_name = args[0]
entity = self.entity_manager.get_entity(entity_name)
logging.debug("EM: %s", self.entity_manager)
logging.debug("Available entities: %s",
self.entity_manager.get_all_entities())
if not entity:
print("Couldn't find the class you are looking for")
return self.help()
if len(args) != 2:
print("You must only provide a single id to destroy")
return
entity = entity.find(args[1])
if entity is None:
print("Couldn't find {} {}".format(args[0], args[2]))
return
id = entity.key
questions = [
inquirer.Confirm(
"confirmed",
message="Are you sure you want to delete {} {}".format(
entity_name, id),
default=False)
]
res = inquirer.prompt(questions)
if not res["confirmed"]:
logging.debug("ABORTED")
return
logging.debug("DELETING ENTITY: %s", entity)
print("Deleting {}:\n\n".format(entity_name))
entity.delete()
print("{} {} deleted succesfully".format(entity_name, id))