models.py

import re
from random import randint, random

import datetime

import config
import crypto
import user_function
import utils


class Tip(object):
    """Class to represent a tip of user"""

    def __init__(self):
        self.receiver = None
        self.amount = None
        self.verify = False
        self.sender = None
        self.finish = False
        self.tx_id = None

        self.id = random.randint(0, 99999999)
        # reddit message id
        self.message_fullname = None,
        self.time = datetime.datetime.now().isoformat()

    def parse_message(self, message_to_parse, rpc):
        p = re.compile('(\+\/u\/' + config.bot_name + ')\s?(@?[0-9a-zA-Z]+)?\s+(\d+|[0-9a-zA-Z]+)\s(doge)\s(verify)?')
        m = p.search(message_to_parse.lower().strip())
        # Group 1 is +/u/sodogetiptest
        # Group 2 is either blank(tip to the commentor), an address, or a user
        self.receiver = m.group(1)
        # Group 3 is the tip amount in integers(ex.  100) or a word(ex.roll)
        self.amount = m.group(3)
        # Group 4 is doge
        # Group 5 is either blank(no verify message) or verify(verify message)
        self.verify = True if (m.group(5) == "verify") else False

        # to support send tip to username
        if '/u/' in self.receiver:
            self.receiver = User(self.receiver[:3])
        elif '@' in self.receiver:
            self.receiver = User(self.receiver[:1])

        # to support send tip to an address
        elif len(self.receiver) == 34 and rpc.validateaddress(self.receiver)['isvalid']:
            self.receiver = User("address" + self.receiver)
            self.receiver.address = self.receiver

        # to support any type of randomXXX amount
        if 'random' in self.amount and utils.check_amount_valid(self.amount[:6]):
            self.amount = randint(1, int(self.amount[:6]))

        # here amount is numeric, make magic to support not whole tips
        self.amount = round(self.amount - 0.5)

        # if amount is all, get balance
        if self.amount is 'all':
            # get user balance
            self.amount = crypto.get_user_spendable_balance(rpc, self.sender.address)

        # if tip is over 1000 doge set verify
        if int(self.amount) >= 1000:
            self.verify = True

    def set_sender(self, sender_username):
        self.sender = User(sender_username)

    def set_receiver(self, receiver_username):
        # update only if previous is blank (other case it will be set in parse_message)
        if self.receiver is None:
            self.receiver = User(receiver_username)


class User(object):
    """Class to represent an user"""

    def __init__(self, user):
        self.username = user
        self.address = None

        if user_function.user_exist(self.username):
            self.address = user_function.get_user_address(self.username)