diff options
author | Lars-Dominik Braun <lars@6xq.net> | 2017-11-22 08:56:25 +0100 |
---|---|---|
committer | Lars-Dominik Braun <lars@6xq.net> | 2017-11-22 08:56:25 +0100 |
commit | 82d779208d208c225e7b3deefc3aa45bb0aedce8 (patch) | |
tree | 1164adb0bfd301b011c172a91e43a0f99cea5a7d /crocoite/cli.py | |
parent | 8c75a5cc52605348df3718ea332e74c15fc0b358 (diff) | |
download | crocoite-82d779208d208c225e7b3deefc3aa45bb0aedce8.tar.gz crocoite-82d779208d208c225e7b3deefc3aa45bb0aedce8.tar.bz2 crocoite-82d779208d208c225e7b3deefc3aa45bb0aedce8.zip |
Add example fixups for Instagram
Diffstat (limited to 'crocoite/cli.py')
-rw-r--r-- | crocoite/cli.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/crocoite/cli.py b/crocoite/cli.py index 54a31d9..9a22d24 100644 --- a/crocoite/cli.py +++ b/crocoite/cli.py @@ -327,7 +327,7 @@ def main (): parser.add_argument('--browser', default='http://127.0.0.1:9222', help='DevTools URL') parser.add_argument('--timeout', default=10, type=int, help='Maximum time for archival') parser.add_argument('--idle-timeout', default=2, type=int, help='Maximum idle seconds (i.e. no requests)', dest='idleTimeout') - parser.add_argument('--onload', default=packageData ('scroll.js'), help='') + parser.add_argument('--onload', action='append', help='') parser.add_argument('--log-buffer', default=1000, type=int, dest='logBuffer') parser.add_argument('--keep-tab', action='store_true', default=False, dest='keepTab', help='Keep tab open') parser.add_argument('url', help='Website URL') @@ -338,8 +338,14 @@ def main (): stopVarname = '__' + __package__ + '_stop__' # avoid sites messing with our scripts by using a random stop variable name newStopVarname = randomString () - with open (args.onload, 'r') as fd: - onload = 'var {} = false;\n'.format (newStopVarname) + fd.read ().replace (stopVarname, newStopVarname) + onload = ['var {} = false;\n'.format (newStopVarname)] + for path in args.onload: + if not os.path.exists (path): + # search for defaults scripts in package data directory + path = packageData (path) + with open (path, 'r') as fd: + onload.append (fd.read ().replace (stopVarname, newStopVarname)) + onload = '\n'.join (onload) stopVarname = newStopVarname # temporary store for requests |