basicswap_miserver/basicswap/ui/page_automation.py

# -*- coding: utf-8 -*-

# Copyright (c) 2022 tecnovert
# Distributed under the MIT software license, see the accompanying
# file LICENSE or http://www.opensource.org/licenses/mit-license.php.

import os

from .util import (
    PAGE_LIMIT,
    get_data_entry,
    have_data_entry,
    set_pagination_filters,
)
from basicswap.util import (
    ensure,
)
from basicswap.db import (
    strConcepts,
)


def page_automation_strategies(self, url_split, post_string):
    server = self.server
    swap_client = server.swap_client

    filters = {
        'page_no': 1,
        'limit': PAGE_LIMIT,
        'sort_by': 'created_at',
        'sort_dir': 'desc',
    }

    messages = []
    form_data = self.checkForm(post_string, 'automationstrategies', messages)

    if form_data and have_data_entry(form_data, 'applyfilters'):
        if have_data_entry(form_data, 'sort_by'):
            sort_by = get_data_entry(form_data, 'sort_by')
            ensure(sort_by in ['created_at', 'rate'], 'Invalid sort by')
            filters['sort_by'] = sort_by
        if have_data_entry(form_data, 'sort_dir'):
            sort_dir = get_data_entry(form_data, 'sort_dir')
            ensure(sort_dir in ['asc', 'desc'], 'Invalid sort dir')
            filters['sort_dir'] = sort_dir

    set_pagination_filters(form_data, filters)

    formatted_strategies = []
    for s in swap_client.listAutomationStrategies(filters):
        formatted_strategies.append((s[0], s[1], strConcepts(s[2])))

    template = server.env.get_template('automation_strategies.html')
    return bytes(template.render(
        title=server.title,
        h2=server.title,
        messages=messages,
        filters=filters,
        strategies=formatted_strategies,
        form_id=os.urandom(8).hex(),
    ), 'UTF-8')


def page_automation_strategy_new(self, url_split, post_string):
    server = self.server
    swap_client = self.server.swap_client

    messages = []
    form_data = self.checkForm(post_string, 'automationstrategynew', messages)

    template = server.env.get_template('automation_strategy_new.html')
    return bytes(template.render(
        title=server.title,
        h2=server.title,
        messages=messages,
        form_id=os.urandom(8).hex(),
    ), 'UTF-8')


def page_automation_strategy(self, url_split, post_string):
    ensure(len(url_split) > 2, 'Strategy ID not specified')
    try:
        strategy_id = int(url_split[2])
    except Exception:
        raise ValueError('Bad strategy ID')

    server = self.server
    swap_client = self.server.swap_client

    messages = []

    strategy = swap_client.getAutomationStrategy(strategy_id)

    formatted_strategy = {
        'label': strategy.label,
        'type': strConcepts(strategy.type_ind),
        'only_known_identities': 'True' if strategy.only_known_identities is True else 'False',
        'data': strategy.data,
        'note': strategy.note,
        'created_at': strategy.created_at,
    }

    template = server.env.get_template('automation_strategy.html')
    return bytes(template.render(
        title=server.title,
        h2=server.title,
        messages=messages,
        strategy=formatted_strategy,
        form_id=os.urandom(8).hex(),
    ), 'UTF-8')