darkfi/bin/tau/tau-python/api.py

import asyncio, json, random
import sys

# import lib.config
from lib.net import Channel

async def create_channel():
    # server_name = lib.config.get("server", "localhost")
    server_name = "localhost"
    reader, writer = await asyncio.open_connection(server_name, 23341)
    channel = Channel(reader, writer)
    return channel

def random_id():
    return random.randint(0, 2**32)

async def query(method, params):
    channel = await create_channel()
    request = {
        "id": random_id(),
        "method": method,
        "params": params,
        "jsonrpc": "2.0",
    }
    await channel.send(request)

    response = await channel.receive()
    # Closed connect returns None
    if response is None:
        print("error: connection with server was closed", file=sys.stderr)
        sys.exit(-1)

    if "error" in response:
        error = response["error"]
        errcode, errmsg = error["code"], error["message"]
        print(f"error: {errcode} - {errmsg}", file=sys.stderr)
        sys.exit(-1)

    return response["result"]

async def get_info():
    return await query("get_info", [])
########
async def get_ref_ids():
    return await query("get_ref_ids", [])

async def get_task_by_ref_id(refid):
    return await query("get_task_by_ref_id", [refid])

########
async def add_task(task):
    return await query("add", [task])

async def fetch_active_tasks():
    return await query("fetch_active_tasks", [])

async def fetch_deactive_tasks(month):
    return await query("fetch_deactive_tasks", [month])

async def fetch_task(task_id):
    return await query("fetch_task", [task_id])

async def fetch_archive_task(task_id, month):
    return await query("fetch_archive_task", [task_id, month])

async def modify_task(who, id, changes):
    return await query("modify_task", [who, id, changes])

async def change_task_status(who, id, status):
    await query("change_task_status", [who, id, status])
    return True

async def add_task_comment(who, id, comment):
    await query("add_task_comment", [who, id, comment])
    return True