add arg handling
This commit is contained in:
parent
f44cc0a4c1
commit
b4ed2381d5
@ -8,7 +8,7 @@ import argparse
|
|||||||
from progress.bar import IncrementalBar
|
from progress.bar import IncrementalBar
|
||||||
|
|
||||||
|
|
||||||
def update_wiki(token_id, token_secret, update_csv=False, update_films_by_year=True, update_films_by_country=True):
|
def update_wiki(token_id, token_secret, update_csv, update_films_by_year, update_films_by_country):
|
||||||
if update_csv:
|
if update_csv:
|
||||||
print("Updating CSV")
|
print("Updating CSV")
|
||||||
update_viewings_csv(token_id, token_secret)
|
update_viewings_csv(token_id, token_secret)
|
||||||
@ -69,6 +69,10 @@ if __name__ == '__main__':
|
|||||||
parser.add_argument('token_id', help='API token ID.')
|
parser.add_argument('token_id', help='API token ID.')
|
||||||
parser.add_argument('token_secret', help='API token secret.')
|
parser.add_argument('token_secret', help='API token secret.')
|
||||||
|
|
||||||
|
parser.add_argument('update_csv', help='Update viewings.csv file, default: True', default=True)
|
||||||
|
parser.add_argument('update_films_by_year', help='Update films by release year page, default: True', default=True)
|
||||||
|
parser.add_argument('update films_by_country', help='Update films by country page, default: True', default=True)
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
update_wiki(args.token_id, args.token_secret)
|
update_wiki(args.token_id, args.token_secret, args.update_csv, args.update_films_by_year, args.update_films_by_country)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user