Sindbad~EG File Manager

from core import app
from flask import session
from flask import session,request
from core.model.UserModel import UserModel

class Auth:

	def check(society_id):
		if session.get('login_'+str(society_id),None):
			return True
		else:
			return False			

	def login(logindata,society_id):
		session['login_'+str(society_id)] = logindata
		session['login'] = logindata 
		return session.get('login_'+str(society_id),None)

	def user(society_id):
		return session.get('login_'+str(society_id),None)

	def logout(society_id):
		session.pop('login_'+str(society_id), None)
		return True


	def setSociety(soceitydata,society_id):
		session['society_'+str(society_id)] = soceitydata 
		return session.get('society_'+str(society_id),None)

	def getSociety(society_id):
		return session.get('society_'+str(society_id),None)
		
	def checkSociety(society_id):
		if Auth.getSociety(society_id):
			if Auth.getSociety(society_id)["society_host"] == request.host:
				set_society = 0
				society = Auth.getSociety(society_id)
		if set_society == 1:
			society = UserModel().setSociety(request.host)
		
		if society:
			msg = ""
			society = Auth.setSociety(society,society_id)
			return	society,msg
		else:
			msg = "Invalid Link"
			return society,msg	

	def checkSocietybyIdKey(society_id,society_key):
		if Auth.getSociety(society_id):
			if Auth.getSociety(society_id)["society_id"] == society_id:
				return True
			else:

				return "Invalid link"
		else:
			society = UserModel().getSociety(society_id,society_key)
			Auth.setSociety(society,society_id)
		return True

app.jinja_env.globals.update(Auth=Auth)