mirror of
https://github.com/rwinkhart/sshyp.git
synced 2026-09-05 16:47:14 -04:00
RandyTheSilly
5d0abb0537
Former-commit-id: 65168e24ed228816d4089241d4086317488fd696 Former-commit-id: 5c46855d45488b2ec5fd8b53040237fc224b45d4
35 lines
1.2 KiB
Python
Executable File
35 lines
1.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import re
|
|
from sys import argv, exit as s_exit
|
|
|
|
# read arguments
|
|
arguments, replacement = argv[1:], None
|
|
|
|
# define replacement text depending on arguments
|
|
if len(arguments) > 0:
|
|
if arguments[0] == 'TMP':
|
|
replacement = 'symlink("/tmp", f"{home}/.config/sshyp/tmp")'
|
|
elif arguments[0] == 'TERMUX':
|
|
replacement = 'symlink("/data/data/com.termux/files/usr/tmp", f"{home}/.config/sshyp/tmp")\n'
|
|
elif arguments[0] == 'LINUX':
|
|
replacement = 'symlink("/dev/shm", f"{home}/.config/sshyp/tmp")'
|
|
else:
|
|
s_exit()
|
|
|
|
targets = (('UNAME-IMPORT-SSHYP', 'sshyp.py', ''), ('UNAME-IMPORT-STWEAK', 'stweak.py', ''),
|
|
('UNAME-TMP', 'stweak.py', replacement))
|
|
|
|
for target in targets:
|
|
# read text from target file
|
|
text, special = open(f"working/{target[1]}", 'r').read(), ''
|
|
# determine text to end regex with
|
|
if target[2] == '':
|
|
regend = '\n'
|
|
else:
|
|
regend = ''
|
|
# compile regex and modify text
|
|
regex = re.compile(f"# PORT START {target[0]}.*?# PORT END {target[0]}{regend}", re.DOTALL)
|
|
new_text = re.sub(regex, target[2], text)
|
|
# write updated text to target file
|
|
open(f"working/{target[1]}", 'w').write(new_text)
|