universal_create_archive_secret.py

from helpers.secret import *
from helpers.universal_secrets import *
import datetime
import os
import sys

secret_expiration_time_years = 2.5
secret_expiration_time_weeks = 52 * secret_expiration_time_years

def get_secret_expiration():
    secret_archive_expiration = datetime.datetime.now() + datetime.timedelta(weeks=secret_expiration_time_weeks) 
    secret_archive_expiration = secret_archive_expiration.strftime('%s')
    print("Expiration time")
    return secret_archive_expiration

def secret_expiration_check(expiration_time):
    current_time = datetime.datetime.now()
    current_time = current_time.strftime('%s')
    if (current_time > expiration_time):
        return True
    else:
        return False

hardcoded_secret = sys.argv[1]

universal_secret_func = universal_secret("archive")
secrets = universal_secret_func.get_secrets()

for secret_name in secrets:
    secret_production = secret(secret_name) ## secret_name
    secret_archive = secret_production.get_secret()

    secret_archive = json.loads(secret_archive)

    print("Adding manually defined secret to archive")
    ## Stores the expiration date in epcoh as the key and secret as the value....
    secret_archive[get_secret_expiration()] = hardcoded_secret

    print("Checking for any expired secrets")
    expired_secrets = []
    for secret_value in secret_archive:
        if (secret_expiration_check(secret_value)):
            ## Remove Expired Secret
            expired_secrets.append(secret_value)

    for secret_value in expired_secrets:
        del secret_archive[secret_value]
    if (len(expired_secrets) > 0):
        print("Removed",str(len(expired_secrets)),"expired secrets")
    else:
        print("No expired secrets to remove...")

    print("Resyncing the secret archive...")
    secret_archive = json.dumps(secret_archive)
    secret_production.set_secret(secret_archive)