darkfi/bin/tau/tau-python/lib/util.py

import random, time
from datetime import timezone, datetime
UTC = timezone.utc

def random_blob_idx():
    return "%030x" % random.randrange(16**30)

def datetime_to_unix(dt):
    return int(time.mktime(dt.timetuple()))
def now():
    return datetime_to_unix(datetime.now(tz=UTC))

def month_to_unix(month=None):
    month_year = month if month is not None else datetime.now(UTC).strftime("%m%y")
    try:
        dt = datetime.strptime(month_year,"%m%y")
        dt = dt.replace(tzinfo=UTC)
        unix = int(dt.timestamp())
    except ValueError:
        print("Error parsing date!")
        exit(-1)
    return unix

def unix_to_datetime(timestamp):
    timestamp = int(timestamp)
    ts = timestamp if timestamp < 1e10 else timestamp/1000
    return datetime.fromtimestamp(ts, UTC)

task_template = {
    "workspace": str,
    "title": str,
    "tags": list,
    "desc": str,
    "owner": str,
    "assign": list,
    "project": list,
    "due": int,
    "rank": float,
    "bounty": int,
    "created_at": int,
    "state": str,
    "events": list,
    "comments": list,
}

def _enforce_task_format(task):
    for attr, val in task.items():
        val_type = task_template[attr]
        if val is None:
            assert val_type == list or attr not in ["created"]
            continue
        assert isinstance(val, val_type)