From 225c0b8202ba6a76aceb69e6061678303cca26c1 Mon Sep 17 00:00:00 2001 From: break27 Date: Thu, 16 Jul 2026 10:46:10 +0800 Subject: [PATCH] update: added 'directory' parameter --- main.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index 192e37a..fecf527 100644 --- a/main.py +++ b/main.py @@ -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():