#!/usr/local/CyberCP/bin/python
import os.path
import sys
import django
sys.path.append('/usr/local/CyberCP')
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "CyberCP.settings")
django.setup()
import argparse
from loginSystem.models import Administrator, ACL
from plogical import hashPassword
from plogical.acl import ACLManager
from packages.models import Package
from baseTemplate.models import version

def main():

    parser = argparse.ArgumentParser(description='Reset admin user password!')
    parser.add_argument('--password', help='New Password')
    args = parser.parse_args()

    adminPass = args.password

    numberOfAdministrator = Administrator.objects.count()
    if numberOfAdministrator == 0:
        ACLManager.createDefaultACLs()
        acl = ACL.objects.get(name='admin')
        token = hashPassword.generateToken('admin', '1234567')

        email = 'usman@cyberpersons.com'
        admin = Administrator(userName="admin", password=hashPassword.hash_password(adminPass), type=1, email=email,
                              firstName="Cyber", lastName="Panel", acl=acl, token=token)
        admin.save()

        vers = version(currentVersion="1.9", build=2)
        vers.save()

        package = Package(admin=admin, packageName="Default", diskSpace=1000,
                          bandwidth=1000, ftpAccounts=1000, dataBases=1000,
                          emailAccounts=1000, allowedDomains=20)
        package.save()
        print("Admin password successfully changed!")
        return 1

    token = hashPassword.generateToken('admin', adminPass)
    admin = Administrator.objects.get(userName="admin")
    admin.password = hashPassword.hash_password(adminPass)
    admin.token = token
    admin.save()

    print("Admin password successfully changed!")

if __name__ == "__main__":
    main()