mirror of
https://github.com/iio612/DEFENDER.git
synced 2026-02-13 11:14:23 +00:00
71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
import argparse
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
from platform import python_version_tuple
|
|
|
|
parser = argparse.ArgumentParser(description="Python Installation Code")
|
|
parser.add_argument('--check-version', action='store_true', help='Check if the python version is ok!')
|
|
parser.add_argument('--install', action='store_true', help='Run the installation')
|
|
args = parser.parse_args()
|
|
|
|
PYTHON_REQUIRED_VERSION = (3, 10, 0)
|
|
PYTHON_SYSTEM_VERSION = tuple(map(int, python_version_tuple()))
|
|
ROOT_PATH = os.getcwd()
|
|
PYENV = Path(f'{ROOT_PATH}/.pyenv/bin/python')
|
|
USER_HOME_DIRECTORY = Path.home()
|
|
SYSTEMD_PATH = Path(USER_HOME_DIRECTORY).joinpath('.config', 'systemd', 'user')
|
|
PY_EXEC = 'defender.py'
|
|
SERVICE_FILE_NAME = 'defender.service'
|
|
|
|
def check_python_requirement():
|
|
if PYTHON_SYSTEM_VERSION < PYTHON_REQUIRED_VERSION:
|
|
raise RuntimeError(f"Your Python Version is not meeting the requirement, System Version: {PYTHON_SYSTEM_VERSION} < Required Version {PYTHON_REQUIRED_VERSION}")
|
|
|
|
def create_service_file():
|
|
|
|
pyenv = PYENV
|
|
systemd_path = SYSTEMD_PATH
|
|
py_exec = PY_EXEC
|
|
service_file_name = SERVICE_FILE_NAME
|
|
|
|
if not Path(systemd_path).exists():
|
|
print("[!] Folder not available")
|
|
sys.exit(1)
|
|
|
|
contain = f'''[Unit]
|
|
Description=Defender IRC Service
|
|
|
|
[Service]
|
|
ExecStart={pyenv} {py_exec}
|
|
WorkingDirectory={ROOT_PATH}
|
|
SyslogIdentifier=Defender
|
|
Restart=on-failure
|
|
|
|
[Install]
|
|
WantedBy=default.target
|
|
'''
|
|
with open(Path(systemd_path).joinpath(service_file_name), "w") as file:
|
|
file.write(contain)
|
|
print('Service file generated with current configuration')
|
|
print('Running IRC Service ...')
|
|
|
|
print(f"#"*24)
|
|
print("Installation complete ...")
|
|
print("If the configuration is correct, then you must see your service connected to your irc server")
|
|
print(f"If any issue, you can see the log file for debug {ROOT_PATH}{os.sep}logs{os.sep}defender.log")
|
|
print(f"#"*24)
|
|
|
|
def main():
|
|
if args.check_version:
|
|
check_python_requirement()
|
|
sys.exit(0)
|
|
|
|
if args.install:
|
|
create_service_file()
|
|
sys.exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|