Example #1
0
    def insert(self) -> 'Manager':
        if isinstance(self._entity, ReadOnlyEntity):
            return self

        sql_elements = self._entity.to_dict()
        sql_elements.__delitem__(self._entity.get_id_name())

        sql = "insert into {} ({}) values ({})".format(
            self._entity.get_table_name(), ', '.join(sql_elements.keys()),
            ', '.join("'" + str(v) + "'" for v in sql_elements.values()))

        db_connect = UseDatabase(dbconfig)
        with db_connect as cursor:
            cursor.execute(sql)
            db_connect.commit()

        return self
Example #2
0
    def update(self) -> 'Manager':
        if isinstance(self._entity, ReadOnlyEntity):
            return self

        sql_elements = self._entity.to_dict()
        element_id = sql_elements[self._entity.get_id_name()]
        sql_elements.__delitem__(self._entity.get_id_name())
        update_elements = []

        for key, value in sql_elements.items():
            update_elements.append("{} = '{}'".format(key, value))

        query = "update {} set {} where {} = {:d}".format(
            self._entity.get_table_name(), ', '.join(update_elements),
            self._entity.get_id_name(), element_id)

        db_connect = UseDatabase(dbconfig)
        with db_connect as cursor:
            cursor.execute(query)

            db_connect.commit()

        return self