utils.py

from cryptography.fernet import Fernet
import sys
import json
import requests
import requests.exceptions

def create_key(key_file):
    key = Fernet.generate_key()
    with open(key_file, 'w') as f:
        f.write(key.decode('ascii'))
    return key

def load_key(key_file):
    try:
        with open(key_file) as f:
            key = f.read()
    except IOError:
        return None
        
    return key

def save_encrypted(filename, key, content):
    cypher = Fernet(key)
    with open(filename, 'w') as f:
        f.write(cypher.encrypt(content.encode('utf-8')).decode('utf-8'))
    

def load_encrypted(filename, key):
    cypher = Fernet(key)
    with open(filename) as f:
        encrypted = f.read()
        return cypher.decrypt(encrypted.encode('utf-8')).decode('utf-8')

def write_string_to_file(filename, string):
    with open(filename, 'w', encoding = "utf-8") as f:
        f.write(string)

def load_from_file(filename):
    with open(filename, encoding = "utf-8") as f:
        return f.read()

def load_from_json(json_path, class_obj):
    objs = []

    with open(json_path, encoding = "utf-8") as f:
        objs_json = json.load(f)
    
    for ob in objs_json:
        objs.append(class_obj(**ob))
    
    return objs
try:
    with open('.gh-credentials') as f:
        ghauth = tuple(f.read().split())
except FileNotFoundError:
    ghauth = ('', '')


def get_gh_picture(ghname):
    if ghname == '':
        return ''
    try:
        rq = requests.get(f'https://api.github.com/users/{ghname}', auth=ghauth)
        if rq.status_code == 404:
            return ''
        else:
            user_data = rq.json()
            return user_data.get('avatar_url', '')
    except requests.exceptions.RequestException:
        return ''