from google.appengine.ext import db
from google.appengine.api import memcache

import random
from hashlib import sha1

class User(db.Model):
        username = db.StringProperty()
        token    = db.StringProperty()
        secret   = db.StringProperty()
        password = db.StringProperty()

        def create(self, username, token, secret):
                result = self.findByUsername(username)
                if result == None:
                        self.username = username
                        self.token = token
                        self.secret = secret
                        self.put()
                        return self
                return result

        def findByUsername(self, username):
                result = db.GqlQuery("SELECT * FROM User WHERE username = :username", username = username)
                if result.count() == 1:
                        return result.get()
                return None

        def setPassword(self, password):
                salt = sha1(str(random.random())).hexdigest()[:5]
                self.password = salt + sha1(salt+password).hexdigest()

        def checkPassword(self, password):
                salt = self.password[:5]
                check = salt + sha1(salt+password).hexdigest()
                return check == self.password