| Server IP : 199.250.214.225 / Your IP : 18.226.181.96 Web Server : Apache System : Linux vps64074.inmotionhosting.com 3.10.0-1160.105.1.vz7.214.3 #1 SMP Tue Jan 9 19:45:01 MSK 2024 x86_64 User : nicngo5 ( 1001) PHP Version : 7.4.33 Disable Function : exec,passthru,shell_exec,system MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : OFF Directory : /opt/tier1adv/bin/ |
Upload File : |
#!/opt/imh-python/bin/python3
"""Moves a domain between /etc/remotedomains and /etc/localdomains"""
from argparse import ArgumentParser
from pathlib import Path
import sys
# addlocaldomain is symlinked to rmlocaldomain
ADDING = 'addlocal' in sys.argv[0]
ADD_PATH = Path('/etc/localdomains') if ADDING else Path('/etc/remotedomains')
RM_PATH = Path('/etc/remotedomains') if ADDING else Path('/etc/localdomains')
def parse_args() -> list[str]:
"""Obtain list of domains from CLI args"""
parser = ArgumentParser(description=f"move a domain to {ADD_PATH}")
parser.add_argument(
'domain',
metavar='DOMAIN',
nargs='+',
help=f'list of domains to move to {ADD_PATH}',
)
args = parser.parse_args()
return args.domain
def remove_domains(domains: list[str]) -> None:
"""Remove list of domains"""
file_lines = RM_PATH.read_text(encoding='utf-8').splitlines()
save = False
for domain in domains:
if domain in file_lines:
print(f"Removing {domain} from {RM_PATH}.")
save = True
else:
print(f"{domain} was not in {RM_PATH}.")
if not save:
return
data = '\n'.join([x for x in file_lines if x not in domains])
save_data(RM_PATH, f"{data}\n")
def add_domains(domains: list[str]) -> None:
"""Add list of domains"""
file_lines = ADD_PATH.read_text(encoding='utf-8').splitlines()
save = False
for domain in domains:
if domain in file_lines:
print(f"{domain} was already in {ADD_PATH}.")
else:
print(f"Adding {domain} to {ADD_PATH}.")
file_lines.append(domain)
save = True
if not save:
return
data = '\n'.join(file_lines)
save_data(ADD_PATH, f"{data}\n")
def save_data(path: Path, data: str) -> None:
"""Save to a {path}.tmp then mv to {path} to avoid race conditions"""
tmp_path = path.parent.joinpath(f"{path.name}.tmp")
tmp_path.write_text(data, encoding='utf-8')
tmp_path.rename(path)
def main():
"""Collect args, remove from local domains, add to remote,
or vice versa if called through addlocaldomain"""
doms = parse_args()
remove_domains(doms)
add_domains(doms)
if __name__ == '__main__':
main()