api.py

import random
import string
import sys
import requests


def __post_request(url, json_data):
    api_url = f"{api_host}/{url}"
    headers = {'X-API-Key': api_key, 'Content-type': 'application/json'}

    req = requests.post(api_url, headers=headers, json=json_data, verify=False)
    req.close()

    try:
        rsp = req.json()
    except:
        sys.exit(f"API {url}: not a valid JSON response")

    if isinstance(rsp, list):
        rsp = rsp[0]

    if not "type" in rsp or not "msg" in rsp:
        sys.exit(
            f"API {url}: got response without type or msg from Mailcow API")

    if rsp['type'] != 'success':
        sys.exit(f"API {url}: {rsp['type']} - {rsp['msg']}")


def add_user(email, name, active, quotum):
    password = ''.join(random.choices(
        string.ascii_letters + string.digits, k=20))

    json_data = {
        'local_part': email.split('@')[0],
        'domain': email.split('@')[1],
        'name': name,
        'quota': str(quotum),
        'password': password,
        'password2': password,
        "active": 1 if active else 0
    }

    __post_request('api/v1/add/mailbox', json_data)

    json_data = {
        'items': [email],
        'attr': {
            'user_acl': [
                "spam_alias",
                # "tls_policy",
                "spam_score",
                "spam_policy",
                # "delimiter_action",
                # "syncjobs",
                # "eas_reset",
                # "quarantine",
                # "sogo_profile_reset",
                # "quarantine_attachments",
                # "quarantine_notification",
                # "app_passwds",
                # "pushover"
            ]
        }
    }

    __post_request('api/v1/edit/user-acl', json_data)


def edit_user(email, active=None, name=None):
    attr = {}
    if (active is not None):
        attr['active'] = 1 if active else 2
    if (name is not None):
        attr['name'] = name

    json_data = {
        'items': [email],
        'attr': attr
    }

    __post_request('api/v1/edit/mailbox', json_data)


def check_user(email):
    url = f"{api_host}/api/v1/get/mailbox/{email}"
    headers = {'X-API-Key': api_key, 'Content-type': 'application/json'}
    req = requests.get(url, headers=headers, verify=False)
    req.close()

    try:
        rsp = req.json()
    except:
        sys.exit("API get/mailbox: not a valid JSON response")

    if not isinstance(rsp, dict):
        sys.exit("API get/mailbox: got response of a wrong type")

    if (not rsp):
        return (False, False, None)

    if 'active_int' not in rsp and rsp['type'] == 'error':
        sys.exit(f"API {url}: {rsp['type']} - {rsp['msg']}")

    return (True, bool(rsp['active_int']), rsp['name'])


def check_api():
    api_url = f"{api_host}/api/v1/get/status/containers"
    headers = {'X-API-Key': api_key, 'Content-type': 'application/json'}

    req = requests.get(api_url, headers=headers, verify=False)
    req.close()
    if req.status_code == 200:
        return True
    return False