Ejemplo n.º 1
0
# writes config to file
# if setting already in file it should be replaced

from core.utils.crypto import AESCipher
from core.utils.key import get as get_key
from core.config import shared as shared_var

crypto = AESCipher(get_key())


def go(file: str, data_dict: dict, encrypted: bool):
    with open(file, 'r+') as _:
        replace_dict = {}
        file_data_dict = {}

        for line in _.readlines():
            if encrypted:
                decrypted_line = crypto.decrypt(line)
            else:
                decrypted_line = line

            try:
                key, value = decrypted_line.split('=')
                file_data_dict[key] = value.strip()
            except ValueError as error_msg:
                # log error or whatever
                pass

        for key, value in data_dict.items():
            if key in file_data_dict:
                if value != file_data_dict[key]:
Ejemplo n.º 2
0
from sys import argv as sys_argv

action = sys_argv[1]
secret = sys_argv[2]

from core.utils.crypto import AESCipher
from core.utils.key import get as get_key


crypto = AESCipher(key=get_key())

if action == 'decrypt':
    print(crypto.decrypt(secret))

else:
    print(crypto.encrypt(secret))