80 lines
2 KiB
Python
80 lines
2 KiB
Python
import argparse
|
|
import pyperclip
|
|
import re
|
|
import sys
|
|
import time
|
|
|
|
from voussoirkit import betterhelp
|
|
from voussoirkit import passwordy
|
|
from voussoirkit import pathclass
|
|
from voussoirkit import pipeable
|
|
|
|
def loop_once(extension, regex=None):
|
|
try:
|
|
text = pyperclip.paste()
|
|
except Exception:
|
|
return
|
|
|
|
text = text.strip()
|
|
|
|
if len(text.split(sep=None, maxsplit=1)) > 1:
|
|
return
|
|
|
|
if 'http://' not in text and 'https://' not in text:
|
|
return
|
|
|
|
if regex and not re.search(regex, text):
|
|
return
|
|
|
|
path = pathclass.Path(passwordy.random_hex(12)).add_extension(extension)
|
|
pyperclip.copy('')
|
|
pipeable.stdout(f'{path.basename} {text}')
|
|
path.write('w', text, encoding='utf-8')
|
|
|
|
def loop_forever(extension, regex):
|
|
pyperclip.copy('')
|
|
while True:
|
|
loop_once(extension=extension, regex=regex)
|
|
time.sleep(1)
|
|
|
|
def watchforlinks_argparse(args):
|
|
pipeable.stderr('Watching clipboard... Press Ctrl+C to stop.')
|
|
try:
|
|
loop_forever(extension=args.extension, regex=args.regex)
|
|
except KeyboardInterrupt:
|
|
pass
|
|
return 0
|
|
|
|
def main(argv):
|
|
parser = argparse.ArgumentParser(
|
|
description='''
|
|
This program will continuously watch your clipboard for http:// and
|
|
https:// URLs and save them to individual files. The files will have
|
|
randomly generated names. The current contents of your clipboard will
|
|
be erased.
|
|
''',
|
|
)
|
|
|
|
parser.add_argument(
|
|
'extension',
|
|
nargs='?',
|
|
type=str,
|
|
default='generic',
|
|
help='''
|
|
The saved files will have this extension.
|
|
''',
|
|
)
|
|
parser.add_argument(
|
|
'--regex',
|
|
type=str,
|
|
default=None,
|
|
help='''
|
|
A regex pattern. Only URLs that match this pattern will be saved.
|
|
''',
|
|
)
|
|
parser.set_defaults(func=watchforlinks_argparse)
|
|
|
|
return betterhelp.go(parser, argv)
|
|
|
|
if __name__ == '__main__':
|
|
raise SystemExit(main(sys.argv[1:]))
|