player.py

# -*- coding: utf8 -*-

import sys
import getpass
import utils
import gender
import species
import character
from Model import Model
import datetime


class player(character.character):
	def __init__(self, login, password):
		self._login = login
		self._password = password
		self._model = None

	def isConnected(self):
		return self._model is not None

	#~ Connect the player by asking him to enter his login and his password
	def loadPlayerFromStdIn(self):
		if self._login is None or self._password is None:
			self._readLoginAndPassword(False, False)

	#~ Method to connect the player
	def connect(self):
		self._model = model.loadByLoginAndPassword(self._login, self._password)

		if len(self._model) == 0:
			self._login = None
			self._password = None
			raise exception("Invalid login or password")

		return True

	#~ Read the login and the password from stdin
	def _readLoginAndPassword(self, checkLogin, confirmPassword):
		while self._login is None or self._login == '':
			self._login = utils.read("Login: ")

			if checkLogin and len(model.loadBy({'login': self._login})):
				print('This login is already used')
				self._login = None

		confirmPassword = ''
		while (self._password is None or self._password == ''):
			self._password = getpass.getpass("Password: ")

			if confirmPassword:
				confirmPassword = getpass.getpass("Confirm password: ")
			else:
				confirmPassword = self._password

			if self._password != confirmPassword:
				print('The passwords do not match')
				self._password = None

	def createNewPlayerFromStdIn(self):
		self._readLoginAndPassword(True, True)

		genders = gender.model.loadAll()
		nbGenders = len(genders)

		for k, v in enumerate(genders):
			print(v['name'] + " (" + str(k) + ")")

		g = -1
		while g < 0 or g >= nbGenders:
			g = utils.read("Character gender: ")
			try:
				g = int(g)
			except:
				g = -1

		genderId = genders[g]['id_gender']

		sps = species.model.getSpecies(genders[g]['name'])
		nbSpecies = len(sps)

		for k, v in enumerate(sps):
			print(v['name'] + " (" + str(k) + ")")
			print(v['description'])

		sp = -1
		while sp < 0 or sp >= nbSpecies:
			sp = utils.read("Character species: ")
			try:
				sp = int(sp)
			except:
				sp = -1

		speciesId = sps[sp]['id_species']

		self._model = {
			'login': self._login,
			'name': self._login,
			'password': self._password,
			'id_species': speciesId,
			'id_gender': genderId,
			'id_area': 1
		}

		self._model['id_character'] = character.model.insert(self._model)
		model.insert(self._model)


class model(character.model):
	fields = ['id_player', 'login', 'password', 'id_character']

	@staticmethod
	def loadByLoginAndPassword(login, password):
		pm = model.loadBy({'login': login, 'password': password})

		if len(pm) == 0:
			return dict()

		pm = pm[0]
		cm = character.model.loadById(pm['id_character'])
		cm.update(pm)

		return cm


class exception(BaseException):
	pass