53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
|
from pymysql import connect
|
||
|
|
from argparse import ArgumentParser
|
||
|
|
from os.path import isfile
|
||
|
|
from json import dump, load
|
||
|
|
from whmcs.get_whmcs import get_whmcs_ipv4, get_whmcs_mac
|
||
|
|
from whmcs.insert_whmcs import insert_whmcs_ipv4
|
||
|
|
from router.insert_router import insert_router_ipv4
|
||
|
|
from ip.ipv4 import ipv4
|
||
|
|
|
||
|
|
if not isfile("config.json"):
|
||
|
|
with open("config.json", "w") as config:
|
||
|
|
data = {"database": {"host": "", "user": "", "password": "", "name": ""}, "ssh": {"host": "", "port": 22, "user": "", "key": ""}, "interface": {"default": ""}}
|
||
|
|
dump(data, config)
|
||
|
|
print("Config file created, please fill it")
|
||
|
|
exit()
|
||
|
|
with open("config.json", "r") as config:
|
||
|
|
conf = load(config)
|
||
|
|
DB_HOST = conf["database"]["host"]
|
||
|
|
DB_USER = conf["database"]["user"]
|
||
|
|
DB_PASS = conf["database"]["password"]
|
||
|
|
DB_NAME = conf["database"]["name"]
|
||
|
|
|
||
|
|
SSH_HOST = conf["ssh"]["host"]
|
||
|
|
SSH_PORT = conf["ssh"]["port"]
|
||
|
|
SSH_USER = conf["ssh"]["user"]
|
||
|
|
SSH_KEY = conf["ssh"]["key"]
|
||
|
|
|
||
|
|
DEFAULT_INTERFACE = conf["interface"]["default"]
|
||
|
|
|
||
|
|
pars = ArgumentParser()
|
||
|
|
pars.add_argument("interface", help="Interface of IPs")
|
||
|
|
pars.add_argument("prefix", help="IPs prefix")
|
||
|
|
pars.add_argument("-d", "--debug", help="Any consequence", action="store_true")
|
||
|
|
args = pars.parse_args()
|
||
|
|
|
||
|
|
debug = False
|
||
|
|
if args.debug:
|
||
|
|
debug = True
|
||
|
|
print("DEBUG MOD ACTICATED !")
|
||
|
|
|
||
|
|
# DB connection
|
||
|
|
db = connect(DB_HOST, DB_USER, DB_PASS, DB_NAME)
|
||
|
|
|
||
|
|
ipl = get_whmcs_ipv4(db)
|
||
|
|
macl = get_whmcs_mac(db)
|
||
|
|
|
||
|
|
# Get list of ip, mac and subnet_mask to add
|
||
|
|
out = ipv4(args.prefix, ipl, macl)
|
||
|
|
|
||
|
|
# Insert the list
|
||
|
|
insert_whmcs_ipv4(out, args.interface, db, debug)
|
||
|
|
insert_router_ipv4(out, "", SSH_HOST, SSH_PORT, SSH_USER, SSH_KEY, debug)
|