models.py

import random

from sqlalchemy import Table, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base


Base = declarative_base()

def gen_hex():
    all = "0123456789abcdef"
    result = [random.choice('abcdef')] + [random.choice(all) for _ in range(4)]
    random.shuffle(result)
    result.insert(0, random.choice(all[1:]))
    return ''.join(result)


class Wallet(Base):

    __tablename__ = 'wallets'

    id          =   Column(Integer, primary_key=True)
    address     =   Column(String(99), unique=True, nullable=False)
    userid      =   Column(Integer, default=-1)
    messageid   =   Column(Integer, default=0)
    deposit     =   Column(String(99), unique=True, nullable=True)

    def __repr__(self):
        return "<User {} has Wallet {}>".format(self.userid,self.address)

    def __init__(self, address, userid, messageid):
        self.address = address
        self.userid = int(userid)
        self.messageid = int(messageid)

class TipJar(Base):

    __tablename__ = 'tips'

    id          =   Column(Integer, primary_key=True)
    paymentid   =   Column(String(64), unique=True, nullable=False)
    userid      =   Column(Integer, default=-1)
    amount      =   Column(Integer, default=0)
    withdraw    =   Column(String(64), unique=True, nullable=False)

    def __repr__(self):
        return "<User {} has {} available to tip>".format(self.userid, self.amount)

    def __init__(self, paymentid, userid, amount):
        self.paymentid = paymentid
        self.userid = int(userid)
        self.amount = int(amount)
        self.withdraw = gen_hex() # use this at the end of PID when withdrawing

class Transaction(Base):

    __tablename__ = 'transactions'

    id          =   Column(Integer, primary_key=True)
    tx          =   Column(String(128), unique=True, nullable=False)
    amount      =   Column(Integer, default=0)
    paymentid   =   Column(String(64), unique=True, nullable=False)

    def __repr__(self):
        return "<tx hash {} transferred {} to PID: {}>".format(self.tx, self.amount, self.paymentid)

    def __init__(self, tx, amount, pid):
        self.tx = tx
        self.amount = amount
        self.paymentid = pid