webHandler.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

from google.appengine.ext import webapp
from google.appengine.ext.webapp import template
from google.appengine.api import xmpp, datastore_errors
from datetime import datetime
from botModels import Acl,Roster,LogsLogin,Sync,InfoMessages
import logging
import md5
import re
import time
import json
from utilities.web import Controller
from utilities.web import parseDateTime
from datetime import timedelta, date



"""
Logowanie do systemu
URL: /login
"""
class LoginHandler(Controller):
	_acl = Acl.GUEST
	_title = 'Logowanie'
	_menu_ac = 'index'
	
	# greeting.content|escape
	def _get(self, params):
		self.view['info'] = ''
		if self._is_logged:
			super(Controller, self).redirect('/')
		elif (self.request.get('hash')):
			try:
				user = Roster.get(self.request.get('hash'))
				if(user != None):
					self.__init_session(user)
					super(Controller, self).redirect('/')
				else:
					self.view['info'] = 'Błędny login lub hasło.'
			except datastore_errors.BadKeyError:
				self.view['info'] = 'Błędny login lub hasło.'
	def __init_session(self,user):
		self.session['auth'] = {
			'jid' : user.jid,
			'password' : user.password,
			'acl' : user.acl
		}
		log = LogsLogin(userRef=user)
		log.put()

"""
Wylogowanie z systemu 
URL: /logout
"""
class LogoutHandler(Controller):
	_acl = Acl.GUEST
	def _get(self, params):
		self.set_no_render(True)
		if self._is_logged:
			self.session.delete()
		super(Controller, self).redirect('/')
	def _post(self):
		self._get()

class ChatsHandler(Controller):
	_acl = Acl.MEMBER
	_title = 'Historia rozmów'
	_menu_ac = 'chats'
	_js = ['underscore.js','backbone.js','chats.js']
	_css = ['chats.css']
	def _get(self, params):
		# Default 
		dt = date.today()
		page = 0
		limit = 100
		items = []
		button_ac = 'dzis'
		frame = {}

		# Params in URL
		if len(params) > 0:
			dt = params[0].lower()
			page = int(params[1]) if (params[1] != None and int(params[1]) > 0) else 0
			if (dt == 'wczoraj'):
				dt = date.today() - timedelta(1)
				button_ac = 'wczoraj'
			elif None != re.match('(\d{4}\-\d{2}\-\d{2})',dt):
				dt = parseDateTime(dt)
				button_ac = 'dzien'
			else:
				dt = date.today()

		startDate = parseDateTime(str(dt)+' 00:00:00')
		endDate = parseDateTime(str(dt)+' 23:59:59')

		chats = InfoMessages.all()
		chats.filter('created >=', startDate)
		chats.filter('created <=', endDate)
		chats.order("-created")
		chats = chats.fetch(limit, limit*page)

		for chat in chats:
			if len(frame) == 0:
				frame = {'author':chat.jid,'messages':[]}
			if frame['author'] == chat.jid:
				frame['messages'].append({'body':chat.message,'time':chat.created.strftime('%Y-%m-%d %H:%M')})
			else:
				items.append(frame)
				frame = {'author':chat.jid,'messages':[]}
				frame['messages'].append({'body':chat.message,'time':chat.created.strftime('%Y-%m-%d %H:%M')})
		if len(frame) > 0:
			items.append(frame)
			frame = {}
		if self.request.get('ajax',False):
			self.set_no_render(True)
			self.response.out.write(json.dumps(items))
		else:
			self.view['chats'] = json.dumps(items)
			self.view['button_ac'] = button_ac

"""
Strona główna
URL: /
"""
class IndexHandler(Controller):
	_title = 'Dashboard'
	_template = 'index'
	_menu_ac = 'index'

	def _get(self, params):
		pass
	def _post(self):
		pass


"""
class SyncHandler(Controller):
	_acl = Acl.MEMBER
	_title = 'Ostatnio przeprowadzone synchronizacje'
	_limit = 10
	
	def _get(self):
		self.view['lastSync'] = []
		self.view['ile'] = self._limit
		
		lastSync = Sync.all()
		lastSync.order('-__key__')
		lastSync = lastSync.fetch(10)
		if len(lastSync) > 0:
			ids = ''
			for item in lastSync:
				ids+=str(item.key().id())+','
			query = SyncAnswers.gql("WHERE syncId IN (%s)" % (ids[0:-1]))
			wynik = query.fetch(100)
			
			odp = {}
			liczniki = {}
			if len(wynik) > 0:
				for i in wynik:
					if False == odp.has_key(int(i.syncId)):
						odp[int(i.syncId)] = []
						liczniki[int(i.syncId)] = {'za':0,'przeciw':0}
					odp[int(i.syncId)].append({'jid':i.jid,'syncCmd':i.syncCmd})
					if(True == i.syncCmd):
						liczniki[int(i.syncId)]['za']+=1
					else:
						liczniki[int(i.syncId)]['przeciw']+=1
			self.view['answers'] = odp
			for item in lastSync:
				if False == liczniki.has_key(int(item.key().id())):
					liczniki[int(item.key().id())] = {'za':0,'przeciw':0}
				self.view['lastSync'].append({
					'id':item.key().id(),
					'jid':item.jid,
					'created':item.created,
					'done':item.done,
					'files':item.files,
					'za' : liczniki[int(item.key().id())]['za'],
					'przeciw' : liczniki[int(item.key().id())]['przeciw']
				})
			self.view['liczniki'] = liczniki
			

class SolutionsHandler(Controller):
	_acl = Acl.MEMBER
	_title = 'Solutions'
	
	def _get(self):
		self.view['solution_list'] = []
		solutions  = Solution.all()
		solutions.order('-__key__')
		solutions = solutions.fetch(50)
		if len(solutions) > 0:
			for i in solutions:
				self.view['solution_list'].append({
					'id' : i.key().id(),
					'title' : i.title
				})
	def _post(self):
		pass
	
class SolutionHandler(Controller):
	_acl = Acl.MEMBER
	_title = 'Solution'
	
	def _get(self):
		id = int(self.request.url.split('/')[-1])
		self.view['solution_id'] = id
		self.view['solution'] = {}
		if id > 0:
			solution = Solution.get_by_id(id)
			if solution != None:
				self.view['solution'] = {
					'title':solution.title,
					'body':solution.body,
					'created':solution.created,
					'jid':solution.author.jid,
					'language':solution.language
				}
				return True
		super(Controller, self).redirect('/solutions')
		return False
	def _post(self):
		pass
	
class SolutionsAddHandler(Controller):
	_acl = Acl.MEMBER
	_title = 'Dodaj solucję'
	_post = {'title':'','body':'','language':2}
	
	def _get(self):
		pass
	def _post(self):
		self.view['errors'] = {}
		if self._validate():
			solution = Solution(title=self._post['title'],body=self._post['body'],author=Roster.findByJid(self.session['auth']['jid']),language=int(self._post['language']))
			solution.put()
			super(Controller, self).redirect('/solutions/%s' % (solution.key().id()))
		self.view['post'] = self._post
		
	def _validate(self):
		title = self.request.get('title')
		body = self.request.get('body')
		language = self.request.get('language')
		
		if len(title) < 5:
			self.view['errors']['title'] = 'Tytuł jest za krótki. Min to 5 znaków.'
		if len(body) < 5:
			self.view['errors']['body'] = 'Kod jest za krótki. Min to 5 znaków.'
		
		self._post = {'title':title,'body':body,'language':language}
		return len(self.view['errors']) == 0
"""