83 lines
3.3 KiB
Python
83 lines
3.3 KiB
Python
import argparse
|
|
|
|
parser = argparse.ArgumentParser(prog='daun',
|
|
description='Dumb Additional Util Nativeier - if you see this but didn\'t download '
|
|
'it, you may have a virus or your friend is motherhacker :)')
|
|
parser.add_argument('--add-path', help='folder to add to path',
|
|
metavar='P:/ath/To/Folder', dest='add_path')
|
|
parser.add_argument('--add-var', help='add variable to environment and its value',
|
|
nargs=2, dest='add_var', metavar=('VAR', 'value'))
|
|
parser.add_argument('--screenshot', help='make a screenshot of all screens, and save it to specified file or to imgur'
|
|
' if no file specified and client id is specified',
|
|
metavar='P:/ath/To/Folder/With/screenshot.jpg', const='imgur', nargs='?')
|
|
parser.add_argument('--imgur', help='imgur app client id',
|
|
metavar='69zxc420228wtf', dest='imgur_client_id')
|
|
parser.add_argument('--wp-control', help='control command for wallpaper engine',
|
|
choices=['pause', 'stop', 'play', 'mute', 'unmute'],
|
|
dest='wp_control')
|
|
parser.add_argument('--set-wallpaper', help='link (web image url or file path) to wallpaper to be set',
|
|
metavar='P:/ath/To/wallpaper.jpg', dest='set_wallpaper')
|
|
parser.add_argument('-d', '--download', help='download file from url to specified',
|
|
nargs=2, metavar=('https://sample.url/to/file.ext', 'P:/ath/To/Folder/With/file.ext'),
|
|
dest='download')
|
|
parser.add_argument('--get-proc-path', help='get path to process by name or PID',
|
|
metavar='daun.exe', dest='get_proc_path')
|
|
parser.add_argument('--kill-proc', help='kill process by PID or name',
|
|
metavar='daun.exe', dest='kill_proc')
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
if args.add_path:
|
|
from modules import path
|
|
path.add_to_path(program_path=args.add_path)
|
|
print('Added {} to path'.format(args.add_path))
|
|
|
|
|
|
if args.add_var:
|
|
from modules import path
|
|
path.add_var(program_path=args.add_var[1], name=args.add_var[0])
|
|
print('Added {0} to environment with value {1}'.format(args.add_var[0], args.add_var[1]))
|
|
|
|
|
|
if args.screenshot:
|
|
from modules import screenshot
|
|
if args.screenshot == 'imgur':
|
|
if args.imgur_client_id:
|
|
print(screenshot.upload_to_imgur(client_id=args.imgur_client_id))
|
|
else:
|
|
print('You need to specify client id to upload to imgur')
|
|
else:
|
|
print(screenshot.save_screenshot(filename=args.screenshot))
|
|
|
|
|
|
if args.wp_control:
|
|
from modules.wallpaperengine import control_we
|
|
control_we(args.wp_control)
|
|
|
|
|
|
if args.set_wallpaper:
|
|
from modules.wallpaper import set_wallpaper
|
|
set_wallpaper(args.set_wallpaper)
|
|
|
|
|
|
if args.download:
|
|
from modules.download import download
|
|
download(args.download[0], args.download[1])
|
|
|
|
|
|
if args.get_proc_path:
|
|
from modules.process import get_location
|
|
try:
|
|
print(get_location(pid=int(args.get_proc_path)))
|
|
except ValueError:
|
|
print(get_location(process_name=args.get_proc_path))
|
|
|
|
|
|
if args.kill_proc:
|
|
from modules.process import kill
|
|
try:
|
|
kill(pid=int(args.kill_proc))
|
|
except ValueError:
|
|
kill(process_name=args.kill_proc)
|