def run(self, *args):
em = EntityManager()
if len(args) < 1:
return self.help()
entity_name = args[0]
entity = em.get_entity(entity_name)
if not entity:
print("Couldn't find the class you are looking for")
return self.help()
ids = args[1:]
res = None
if len(ids) > 0:
res = entity.find(ids)
else:
res = entity.find()
data = []
self.headers = list(entity.fields.keys())
self.headers.insert(0, "id")
logging.debug("GOT MODELS: %s",res)
if isinstance(res, Entity):
data = [e.key] + [ self._get_data_from_entity(res) ]
else:
for e in res:
data.append([e.key] + self._get_data_from_entity(e))
if len(data) <= 0:
data = [ [ "" for i in range(len(self.headers)) ]]
logging.debug(data)
logging.debug(self.headers)
string = tt.to_string(
data,
header=self.headers,
style=tt.styles.ascii_thin_double,
)
print(string)
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))
def parent_entity(self):
em = EntityManager()
parent = em.get_entity(self.parent_entity)
return parent