user.py

import email.mime.text
import uuid
import multiprocessing.pool

import tornado.gen

import settings


class User(object):
    collection = settings.db['users']

    def __init__(self, login, **kwargs):
        self.id = kwargs.get('_id')
        self.login = login
        self.data = kwargs.get('data')
        self.token = kwargs.get('token', str(uuid.uuid4()))
        self.gh_token = kwargs.get('gh_token')

    def __lt__(self, other):
        return self.login < other.login

    @property
    def admin(self):
        return self.login in settings.ADMINS

    @classmethod
    @tornado.gen.coroutine
    def get(cls, token=None, login=None):
        query = {}
        if token:
            query['token'] = token
        if login:
            query['login'] = login

        if not query:
            return None

        user = yield User.collection.find_one(query)
        return cls(**user) if user else None

    @tornado.gen.coroutine
    def put(self):
        saved = self.id is not None
        user = {
            'login': self.login,
            'data': self.data,
            'token': self.token,
            'gh_token': self.gh_token,
        }

        if saved:
            user['_id'] = self.id

        self.id = yield User.collection.save(user)