Skip to content

Commit 330dfcd

Browse files
committed
Refactor code to include --imports option in main.py
1 parent f9e5ef1 commit 330dfcd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

main.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ async def main():
105105

106106
if args.imports == 'ind' or args.imports == 'all':
107107
collection_tasks = [col.get('function')(url, col, headers) for col in collections if col.get('method') != 'ExportPhotos']
108-
elif args.imports == 'pictures' or args.imports == 'all':
108+
if args.imports == 'pictures' or args.imports == 'all':
109109
collection_tasks = [col.get('function')(url, col, headers) for col in collections if col.get('method') == 'ExportPhotos']
110110

111111
await asyncio.gather(*collection_tasks)
@@ -116,7 +116,7 @@ async def main():
116116
start_time = datetime.now()
117117
if (args.imports == 'ind' or args.imports == 'all'):
118118
await import_ind()
119-
elif (args.imports == 'pictures' or args.imports == 'all'):
119+
if (args.imports == 'pictures' or args.imports == 'all'):
120120
await import_pictures()
121121
end_time = datetime.now()
122122
execution_time = end_time - start_time

0 commit comments

Comments
 (0)