3wordchant/hamster-tools
0
0
Fork 0
Code Issues 8 Pull Requests Releases Wiki Activity

Run autopep8

Browse Source
This commit is contained in:
3wc 2022-11-11 16:43:30 -08:00
parent 503b35c12c
commit 900fb9f0c0
1 changed files with 55 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

90
hamstertools/__init__.py
View File

@ -94,7 +94,7 @@ def list_categories(search):
results = get_categories(search=search) results = get_categories(search=search)
for r in results: for r in results:
click.echo('@{0[0]}: {0[1]}'.format(r)) click.echo('@{0[0]}: {0[1]}'.format(r))
@categories.command('delete') @categories.command('delete')
@ -129,8 +129,8 @@ def delete_categories(ids):
@categories.command('rename') @categories.command('rename')
@click.argument('id_', metavar='ID') @click.argument('id_', metavar='ID')
@click.argument('name') @click.argument('name')
def rename_category(id_, name): def rename_category(id_, name):
""" Rename a category """ """ Rename a category """
@ -163,10 +163,10 @@ def list_category_activities(ids):
seq=','.join(['?'] * len(ids)) seq=','.join(['?'] * len(ids))
) )
results = c.execute(sql, ids) results = c.execute(sql, ids)
for r in results: for r in results:
click.echo('@{0[0]}: {0[2]} » {0[1]}'.format(r)) click.echo('@{0[0]}: {0[2]} » {0[1]}'.format(r))
@cli.group() @cli.group()
@ -231,7 +231,7 @@ def move(category_id, ids):
click.secho('Moving to "@{0[0]}: {0[1]}":'.format(category), fg='green') click.secho('Moving to "@{0[0]}: {0[1]}":'.format(category), fg='green')
for r in results: for r in results:
click.secho('@{0[3]}: {0[2]} » @{0[0]}: {0[1]}'.format(r), fg='blue') click.secho('@{0[3]}: {0[2]} » @{0[0]}: {0[1]}'.format(r), fg='blue')
click.confirm('Do you want to continue?', abort=True) click.confirm('Do you want to continue?', abort=True)
@ -281,7 +281,8 @@ def list_facts(ids):
results = c.execute(sql, (r[0],)) results = c.execute(sql, (r[0],))
for r in results: for r in results:
click.secho('@{0[0]}, {0[1]}'.format(r), fg='blue') click.secho('@{0[0]}, {0[1]}'.format(r), fg='blue')
@activities.command() @activities.command()
@click.argument('from_id') @click.argument('from_id')
@ -314,7 +315,7 @@ def move_facts(from_id, to_id):
results = c.execute(sql, (from_id,)) results = c.execute(sql, (from_id,))
for r in results: for r in results:
click.secho('@{0[0]}, {0[1]}'.format(r), fg='blue') click.secho('@{0[0]}, {0[1]}'.format(r), fg='blue')
click.confirm('Do you want to continue?', abort=True) click.confirm('Do you want to continue?', abort=True)
@ -328,7 +329,8 @@ def move_facts(from_id, to_id):
click.secho('Moved {0} facts'.format(results.rowcount), fg='green') click.secho('Moved {0} facts'.format(results.rowcount), fg='green')
click.confirm( click.confirm(
'Would you like to delete @{0[2]} » @{0[0]}: {0[1]}?'.format(from_activity), 'Would you like to delete @{0[2]} » @{0[0]}: {0[1]}?'.format(
from_activity),
abort=True abort=True
) )
@ -361,7 +363,8 @@ def find_duplicates():
results = c.execute(sql) results = c.execute(sql)
for r in results: for r in results:
click.secho('@{0[0]}: {0[1]} » @{0[2]}: {0[3]} ({0[4]})'.format(r), fg='blue') click.secho(
'@{0[0]}: {0[1]} » @{0[2]}: {0[3]} ({0[4]})'.format(r), fg='blue')
@cli.group() @cli.group()
@ -431,9 +434,12 @@ def sync(username, api_key, just_errors, ignore_activities, mapping_path=None):
'X-AUTH-TOKEN': api_key 'X-AUTH-TOKEN': api_key
} }
customers = requests.get(f'{kimai_api_url}/customers?visible=3', headers=auth_headers).json() customers = requests.get(
projects = requests.get(f'{kimai_api_url}/projects?visible=3', headers=auth_headers).json() f'{kimai_api_url}/customers?visible=3', headers=auth_headers).json()
activities = requests.get(f'{kimai_api_url}/activities?visible=3', headers=auth_headers).json() projects = requests.get(
f'{kimai_api_url}/projects?visible=3', headers=auth_headers).json()
activities = requests.get(
f'{kimai_api_url}/activities?visible=3', headers=auth_headers).json()
found_customers = [] found_customers = []
found_projects = [] found_projects = []
@ -442,38 +448,47 @@ def sync(username, api_key, just_errors, ignore_activities, mapping_path=None):
for row in mapping_data: for row in mapping_data:
# Check if each mapping still exists in Kimai # Check if each mapping still exists in Kimai
matching_customers = list(filter(lambda x: x['name'] == row[0], customers)) matching_customers = list(
filter(lambda x: x['name'] == row[0], customers))
if row[0] in found_customers: if row[0] in found_customers:
just_errors or click.secho("Skipping existing customer '{0}'".format(row[0]), fg='green') just_errors or click.secho(
"Skipping existing customer '{0}'".format(row[0]), fg='green')
else: else:
if len(matching_customers) > 1: if len(matching_customers) > 1:
click.secho("More than one match for customer '{0}'".format(row[0]), fg='red') click.secho(
"More than one match for customer '{0}'".format(row[0]), fg='red')
continue continue
elif len(matching_customers) < 1: elif len(matching_customers) < 1:
click.secho("Missing customer '{0}'".format(row[0]), fg='yellow') click.secho("Missing customer '{0}'".format(
row[0]), fg='yellow')
continue continue
else: else:
just_errors or click.secho("Found customer '{0}'".format(row[0]), fg='green') just_errors or click.secho(
"Found customer '{0}'".format(row[0]), fg='green')
found_customers.append(row[0]) found_customers.append(row[0])
project_str = ':'.join(row[0:2]) project_str = ':'.join(row[0:2])
matching_projects = list(filter( matching_projects = list(filter(
lambda x: x['name'] == row[1] and lambda x: x['name'] == row[1] and
x['customer'] == matching_customers[0]['id'], x['customer'] == matching_customers[0]['id'],
projects) projects)
) )
if project_str in found_projects: if project_str in found_projects:
just_errors or click.secho("Skipping existing project '{0}'".format(project_str), fg='green') just_errors or click.secho(
"Skipping existing project '{0}'".format(project_str), fg='green')
else: else:
if len(matching_projects) > 1: if len(matching_projects) > 1:
click.secho("More than one match for project '{0}'".format(project_str), fg='red') click.secho("More than one match for project '{0}'".format(
project_str), fg='red')
continue continue
elif len(matching_projects) < 1: elif len(matching_projects) < 1:
click.secho("Missing project '{0}'".format(project_str), fg='yellow') click.secho("Missing project '{0}'".format(
project_str), fg='yellow')
continue continue
else: else:
just_errors or click.secho("Found project '{0}'".format(project_str), fg='green') just_errors or click.secho(
"Found project '{0}'".format(project_str), fg='green')
found_projects.append(project_str) found_projects.append(project_str)
if ignore_activities: if ignore_activities:
@ -481,20 +496,24 @@ def sync(username, api_key, just_errors, ignore_activities, mapping_path=None):
activity_str = ':'.join(row) activity_str = ':'.join(row)
if activity_str in found_activities: if activity_str in found_activities:
just_errors or click.secho("Skipping existing activity '{0}'".format(activity_str), fg='green') just_errors or click.secho(
"Skipping existing activity '{0}'".format(activity_str), fg='green')
else: else:
matching_activities = list(filter( matching_activities = list(filter(
lambda x: x['name'] == row[2] lambda x: x['name'] == row[2]
and x['project'] == matching_projects[0]['id'], and x['project'] == matching_projects[0]['id'],
activities activities
)) ))
if len(matching_activities) > 1: if len(matching_activities) > 1:
click.secho("More than one match for activity '{0}'".format(activity_str), fg='red') click.secho("More than one match for activity '{0}'".format(
activity_str), fg='red')
elif len(matching_activities) < 1: elif len(matching_activities) < 1:
click.secho("Missing activity '{0}'".format(activity_str), fg='yellow') click.secho("Missing activity '{0}'".format(
activity_str), fg='yellow')
else: else:
just_errors or click.secho("Found activity '{0}'".format(activity_str), fg='green') just_errors or click.secho(
"Found activity '{0}'".format(activity_str), fg='green')
found_activities.append(activity_str) found_activities.append(activity_str)
@ -587,7 +606,8 @@ def _import(username, mapping_path=None, output=None, category_search=None, afte
except KeyError: except KeyError:
if show_missing: if show_missing:
output_writer.writerow([fact[6], fact[5]]) output_writer.writerow([fact[6], fact[5]])
click.secho("Can't find mapping for '{0}', skipping".format(k), fg='yellow') click.secho(
"Can't find mapping for '{0}', skipping".format(k), fg='yellow')
continue continue
if show_missing: if show_missing:
@ -611,18 +631,18 @@ def _import(username, mapping_path=None, output=None, category_search=None, afte
output_writer.writerow([ output_writer.writerow([
date_start.strftime('%Y-%m-%d'), date_start.strftime('%Y-%m-%d'),
date_start.strftime('%H:%M'), date_start.strftime('%H:%M'),
'', # To (time) '', # To (time)
duration, duration,
'', # Rate '', # Rate
username, username,
mapping_[0], mapping_[0],
mapping_[1], mapping_[1],
mapping_[2], mapping_[2],
fact[3] or '', fact[3] or '',
'0', # Exported '0', # Exported
mapping_[3], mapping_[3],
'', # Hourly rate '', # Hourly rate
'', # Fixed rate '', # Fixed rate
]) ])
output_file.close() output_file.close()