update: added 'directory' parameter
This commit is contained in:
@@ -30,6 +30,7 @@ parser = argparse.ArgumentParser(description="Mailer")
|
||||
parser.add_argument('account', type=str)
|
||||
parser.add_argument('password', type=str)
|
||||
parser.add_argument('-c', '--column', nargs='+', action='extend')
|
||||
parser.add_argument('-d', '--directory', type=str, default=str(Path.home().joinpath('Downloads')))
|
||||
parser.add_argument('-t', '--timeout', type=int, default=60)
|
||||
parser.add_argument('-i', '--interval', type=int, default=10)
|
||||
parser.add_argument('-r', '--attempts', type=int, default=3)
|
||||
@@ -328,11 +329,11 @@ def main(driver: Chrome, logger = logging.getLogger('main')):
|
||||
|
||||
def save():
|
||||
try:
|
||||
file = Path('.').joinpath(filename or f'Mailer-{datetime.now().strftime('%Y%m%d-%H%M%S-%f')}.xlsx').resolve()
|
||||
logger.info('Saving file at %s', str(file))
|
||||
file = Path(parameters['directory']).joinpath(filename or f'Mailer-{datetime.now().strftime('%Y%m%d-%H%M%S-%f')}.xlsx').resolve()
|
||||
logger.info('Saving document at %s', str(file))
|
||||
wb.save(file)
|
||||
except Exception as e:
|
||||
logger.error('Error while writing data to file', exc_info=e)
|
||||
logger.error('Error while writing data', exc_info=e)
|
||||
|
||||
exit = sp.pop('exit')
|
||||
sp.set('exit', lambda: (save() if options.get('writable') and status.value >= 2 else None, exit()))
|
||||
@@ -589,6 +590,7 @@ if __name__ == '__main__':
|
||||
"url": WEBMAIL,
|
||||
}],
|
||||
"profile.default_content_setting_values.notifications": 2,
|
||||
'download.default_directory': args.directory,
|
||||
})
|
||||
|
||||
with keep.presenting():
|
||||
|
||||
Reference in New Issue
Block a user