def pending_list_show(options):
with Session() as session:
try:
pending_list = plugin_pending_list.get_list_by_exact_name(options.list_name, session=session)
except NoResultFound:
console('Could not find pending list with name {}'.format(options.list_name))
return
try:
entry = plugin_pending_list.get_entry_by_id(pending_list.id, int(options.entry), session=session)
except NoResultFound:
console('Could not find matching pending entry with ID {} in list `{}`'.format(int(options.entry),
options.list_name))
return
except ValueError:
entry = plugin_pending_list.get_entry_by_title(pending_list.id, options.entry, session=session)
if not entry:
console('Could not find matching pending entry with title `{}` in list `{}`'.format(options.entry,
options.list_name))
return
header = ['Field name', 'Value']
table_data = [header]
for k, v in sorted(entry.entry.items()):
table_data.append([k, str(v)])
table = TerminalTable(options.table_type, table_data, wrap_columns=[1])
table.table.justify_columns[0] = 'center'
try:
console(table.output)
except TerminalTableError as e:
console('ERROR: %s' % str(e))
def pending_list_add(options):
with Session() as session:
try:
pending_list = plugin_pending_list.get_list_by_exact_name(options.list_name, session=session)
except NoResultFound:
console('Could not find a pending list with name `{}`, creating'.format(options.list_name))
pending_list = plugin_pending_list.PendingListList(name=options.list_name)
session.add(pending_list)
session.merge(pending_list)
session.commit()
title = options.entry_title
entry = {'title': options.entry_title, 'url': options.url}
db_entry = plugin_pending_list.get_entry_by_title(list_id=pending_list.id, title=title, session=session)
if db_entry:
console("Entry with the title `{}` already exist with list `{}`. Will replace identifiers if given".format(
title, pending_list.name))
operation = 'updated'
else:
console("Adding entry with title `{}` to list `{}`".format(title, pending_list.name))
db_entry = plugin_pending_list.PendingListEntry(entry=entry, pending_list_id=pending_list.id)
if options.approved:
console('marking entry as approved')
db_entry.approved = True
session.add(db_entry)
operation = 'added'
if options.attributes:
console('Adding attributes to entry `{}`'.format(title))
for identifier in options.attributes:
for k, v in identifier.items():
entry[k] = v
db_entry.entry = entry
console('Successfully {} entry `{}` to pending list `{}` '.format(operation, title, pending_list.name))
def pending_list_approve(options, approve=None):
with Session() as session:
try:
entry_list = plugin_pending_list.get_list_by_exact_name(options.list_name)
except NoResultFound:
console('Could not find pending list with name `{}`'.format(options.list_name))
return
try:
db_entry = plugin_pending_list.get_entry_by_id(entry_list.id, int(options.entry), session=session)
except NoResultFound:
console('Could not find matching entry with ID {} in list `{}`'.format(int(options.entry),
options.list_name))
return
except ValueError:
db_entry = plugin_pending_list.get_entry_by_title(entry_list.id, options.entry, session=session)
if not db_entry:
console('Could not find matching entry with title `{}` in list `{}`'.format(options.entry,
options.list_name))
return
approve_text = 'approved' if approve else 'rejected'
if (db_entry.approved is True and approve is True) or (db_entry.approved is False and approve is False):
console('entry {} is already {}'.format(db_entry.title, approve_text))
return
db_entry.approved = approve
console('Successfully marked pending entry {} as {}'.format(db_entry.title, approve_text))
def pending_list_approve(options, approve=None):
with Session() as session:
try:
entry_list = get_list_by_exact_name(options.list_name)
except NoResultFound:
console('Could not find pending list with name `{}`'.format(options.list_name))
return
try:
db_entry = get_entry_by_id(entry_list.id, int(options.entry), session=session)
except NoResultFound:
console('Could not find matching entry with ID {} in list `{}`'.format(int(options.entry),
options.list_name))
return
except ValueError:
db_entry = get_entry_by_title(entry_list.id, options.entry, session=session)
if not db_entry:
console('Could not find matching entry with title `{}` in list `{}`'.format(options.entry,
options.list_name))
return
approve_text = 'approved' if approve else 'rejected'
if (db_entry.approved is True and approve is True) or (db_entry.approved is False and approve is False):
console('entry {} is already {}'.format(db_entry.title, approve_text))
return
db_entry.approved = approve
console('Successfully marked pending entry {} as {}'.format(db_entry.title, approve_text))
def pending_list_show(options):
with Session() as session:
try:
pending_list = get_list_by_exact_name(options.list_name, session=session)
except NoResultFound:
console('Could not find pending list with name {}'.format(options.list_name))
return
try:
entry = get_entry_by_id(pending_list.id, int(options.entry), session=session)
except NoResultFound:
console('Could not find matching pending entry with ID {} in list `{}`'.format(int(options.entry),
options.list_name))
return
except ValueError:
entry = get_entry_by_title(pending_list.id, options.entry, session=session)
if not entry:
console('Could not find matching pending entry with title `{}` in list `{}`'.format(options.entry,
options.list_name))
return
header = ['Field name', 'Value']
table_data = [header]
for k, v in sorted(entry.entry.items()):
table_data.append([k, str(v)])
table = TerminalTable(options.table_type, table_data, wrap_columns=[1])
table.table.justify_columns[0] = 'center'
try:
console(table.output)
except TerminalTableError as e:
console('ERROR: %s' % str(e))
def pending_list_add(options):
with Session() as session:
try:
pending_list = get_list_by_exact_name(options.list_name, session=session)
except NoResultFound:
console('Could not find a pending list with name `{}`, creating'.format(options.list_name))
pending_list = PendingListList(name=options.list_name)
session.add(pending_list)
session.merge(pending_list)
session.commit()
title = options.entry_title
entry = {'title': options.entry_title, 'url': options.url}
db_entry = get_entry_by_title(list_id=pending_list.id, title=title, session=session)
if db_entry:
console("Entry with the title `{}` already exist with list `{}`. Will replace identifiers if given".format(
title, pending_list.name))
operation = 'updated'
else:
console("Adding entry with title `{}` to list `{}`".format(title, pending_list.name))
db_entry = PendingListEntry(entry=entry, pending_list_id=pending_list.id)
if options.approved:
console('marking entry as approved')
db_entry.approved = True
session.add(db_entry)
operation = 'added'
if options.attributes:
console('Adding attributes to entry `{}`'.format(title))
for identifier in options.attributes:
for k, v in identifier.items():
entry[k] = v
db_entry.entry = entry
console('Successfully {} entry `{}` to pending list `{}` '.format(operation, title, pending_list.name))
def pending_list_purge(options):
with Session() as session:
try:
entry_list = plugin_pending_list.get_list_by_exact_name(options.list_name)
except NoResultFound:
console('Could not find entry list with name `{}`'.format(options.list_name))
return
console('Deleting list {}'.format(options.list_name))
session.delete(entry_list)
def pending_list_purge(options):
with Session() as session:
try:
entry_list = get_list_by_exact_name(options.list_name)
except NoResultFound:
console('Could not find entry list with name `{}`'.format(options.list_name))
return
console('Deleting list {}'.format(options.list_name))
session.delete(entry_list)
def post(self, session=None):
""" Create a new pending list """
data = request.json
name = data.get('name')
try:
get_list_by_exact_name(name=name, session=session)
except NoResultFound:
pass
else:
raise Conflict('list with name \'%s\' already exists' % name)
pending_list = PendingListList()
pending_list.name = name
session.add(pending_list)
session.commit()
resp = jsonify(pending_list.to_dict())
resp.status_code = 201
return resp
def post(self, session=None):
""" Create a new pending list """
data = request.json
name = data.get('name')
try:
get_list_by_exact_name(name=name, session=session)
except NoResultFound:
pass
else:
raise Conflict('list with name \'%s\' already exists' % name)
pending_list = PendingListList()
pending_list.name = name
session.add(pending_list)
session.commit()
resp = jsonify(pending_list.to_dict())
resp.status_code = 201
return resp
def pending_list_list(options):
"""List pending list entries"""
with Session() as session:
try:
pending_list = get_list_by_exact_name(options.list_name, session=session)
except NoResultFound:
console('Could not find pending list with name `{}`'.format(options.list_name))
return
header = ['#', 'Title', '# of fields', 'Approved']
table_data = [header]
for entry in get_entries_by_list_id(pending_list.id, order_by='added', descending=True, session=session):
approved = colorize('green', entry.approved) if entry.approved else entry.approved
table_data.append([entry.id, entry.title, len(entry.entry), approved])
table = TerminalTable(options.table_type, table_data)
try:
console(table.output)
except TerminalTableError as e:
console('ERROR: %s' % str(e))
def pending_list_del(options):
with Session() as session:
try:
entry_list = plugin_pending_list.get_list_by_exact_name(options.list_name)
except NoResultFound:
console('Could not find pending list with name `{}`'.format(options.list_name))
return
try:
db_entry = plugin_pending_list.get_entry_by_id(entry_list.id, int(options.entry), session=session)
except NoResultFound:
console(
'Could not find matching entry with ID {} in list `{}`'.format(int(options.entry),
options.list_name))
return
except ValueError:
db_entry = plugin_pending_list.get_entry_by_title(entry_list.id, options.entry, session=session)
if not db_entry:
console(
'Could not find matching entry with title `{}` in list `{}`'.format(options.entry,
options.list_name))
return
console('Removing entry `{}` from list {}'.format(db_entry.title, options.list_name))
session.delete(db_entry)
def pending_list_del(options):
with Session() as session:
try:
entry_list = get_list_by_exact_name(options.list_name)
except NoResultFound:
console('Could not find pending list with name `{}`'.format(options.list_name))
return
try:
db_entry = get_entry_by_id(entry_list.id, int(options.entry), session=session)
except NoResultFound:
console(
'Could not find matching entry with ID {} in list `{}`'.format(int(options.entry),
options.list_name))
return
except ValueError:
db_entry = get_entry_by_title(entry_list.id, options.entry, session=session)
if not db_entry:
console(
'Could not find matching entry with title `{}` in list `{}`'.format(options.entry,
options.list_name))
return
console('Removing entry `{}` from list {}'.format(db_entry.title, options.list_name))
session.delete(db_entry)
