"""
User class for MURSAT1 Mission Dashboard
"""
import hashlib
import smtplib
import time
from cgi import parse_qs, escape
from MmdDb import Db
from MmdLocationList import LocationList

class User:

  def __init__ (self, email = 'Anonymous'):
    self.db = Db ()
    self.email = email
    self.location_list = LocationList ()

  def create (self, user_id, firstname, lastname, email, callsign, password):
    password_hash = hashlib.sha1 (password).hexdigest ()
    code =  hashlib.sha1 ('{0}{1}{2}{3}'.format (user_id, firstname, lastname, email)).hexdigest ()
    regtimeout = int (time.strftime ('%s')) + (3600 * 24)
    if not self.db.userCreate (user_id, firstname, lastname, email, callsign, password_hash, code, regtimeout):
      return False
    self.email = email
    self.firstname = firstname
    self.lastname = lastname
    self.user_id = user_id
    return code

  def addLocation (self, name, longitude, latitude, altitude, is_default):
    self.location_list = LocationList (self.user_id)
    return self.location_list.addLocation (name, longitude, latitude, altitude, is_default)

  def findPending (self, code):
    u = self.db.userFindPending (code)
    if not u:
      return False
    if int (u['regtimeout']) < int (time.strftime ('%s')):
      return False
    return True

  def confirm (self, code):
    u = self.db.userFindPending (code)
    if not u:
      return False
    if int (u['regtimeout']) < int (time.strftime ('%s')):
      return False
    self.db.userConfirm (code)
    self.email = u['email']
    self.firstname = u['firstname']
    self.lastname = u['lastname']
    return True

  def load (self, email = False):
    if email:
      self.email = email
    u = self.db.userFindEmail (self.email)
    if not u:
      return False
    self.user_id = u['id']
    self.firstname = u['firstname']
    self.lastname = u['lastname']
    self.password = u['password']
    self.location_list = LocationList (self.user_id)
    return True

  def checkPassword (self, password):
    if hashlib.sha1 (password).hexdigest () == self.password:
      return True
    return False

  def cancel (self, code):
    self.deleteLocationList ()
    self.db.userCancel (code)

  def deleteLocationList (self):
    self.location_list.delete ()

  def getDefaultLocation (self):
    return self.location_list.getDefaultLocation ()

  def sendEmail (self, message):
    server = smtplib.SMTP ('localhost')
    try:
      server.sendmail ('noreply@mur.at', self.email, message)
      server.quit ()
      return True
    except (smtplib.SMTPRecipientsRefused, smtplib.SMTPHeloError, smtplib.SMTPSenderRefused, smtplib.SMTPDataError):
      server.quit ()
      return False

if __name__ == "__main__":
  user = User ()
  email = 'jfh@mur.at'
  try:
    assert user, 'no user object'
    code = user.create (1, 'jogi', 'hofmueller', email, '', 'blah')
    assert code, 'could not create user'
    assert user.load (email), 'could not load user data from db'
    print 'User:', user.firstname, user.lastname, user.email
    assert user.addLocation ('Graz', '14.4', '47.2', '376', 'yes'), 'could not add location'
    location = user.getDefaultLocation ()
    assert location, 'could not get default location'
    print 'Location:', location.name, location.longitude, location.latitude, location.is_default
    user.cancel (code)
    print 'passed all tests and removed objects from database'
  except AssertionError as error:
    print 'Test failed: {0}'.format (error)

# vim: tw=0 ts=2 expandtab
# EOF