106 lines
3 KiB
Python
106 lines
3 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import json
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
import MySQLdb
|
|
|
|
class DB(object):
|
|
|
|
def __init__(self,host,user,pwd,db):
|
|
self.host = host
|
|
self.user = user
|
|
self.pwd = pwd
|
|
self.db = db
|
|
self.con = 0
|
|
|
|
def connect(self):
|
|
if self.con == 0:
|
|
try:
|
|
con = MySQLdb.connect(self.host,self.user,self.pwd,self.db)
|
|
self.con = con
|
|
return True
|
|
except Exception, e:
|
|
log.fatal('Error connecting to database : %s' % e)
|
|
return
|
|
|
|
def do_sql(self,sql):
|
|
try:
|
|
c=self.con.cursor()
|
|
c.execute(sql)
|
|
self.con.commit()
|
|
return c
|
|
except Exception,e:
|
|
log.error('Error executing request %s : %s' % (sql,e))
|
|
return False
|
|
|
|
def select(self,sql):
|
|
ret=self.do_sql(sql)
|
|
if ret!=False:
|
|
return ret.fetchall()
|
|
return ret
|
|
|
|
def login(self,email,password):
|
|
ret=self.select("SELECT email,name,password FROM users WHERE email='%s' AND password='%s'" % (email,password))
|
|
log.debug(ret)
|
|
if ret:
|
|
if len(ret)==1:
|
|
return {
|
|
'email': ret[0][0],
|
|
'name': ret[0][1]
|
|
}
|
|
elif len(ret)>=1:
|
|
log.warning('Duplicate user %s in database' % email)
|
|
elif ret==():
|
|
return { 'loginerror': 'Utilisateur inconnu' }
|
|
return { 'loginerror': 'Erreur inconnu' }
|
|
|
|
def subscribe(self,email,name,password):
|
|
ret=self.select("SELECT count(*) as count FROM users WHERE email='%s'" % (email))
|
|
log.debug(ret)
|
|
if ret[0][0]!=0:
|
|
return {'subscribeerror': u'Cette adresse mail est déjà associés a un compte !'}
|
|
else:
|
|
if self.do_sql("INSERT INTO users (email,name,password) VALUES ('%s','%s','%s')" % (email,name,password)):
|
|
return {
|
|
'email': email,
|
|
'name': name,
|
|
'password': password
|
|
}
|
|
return {'subscribeerror': u'Une erreur est survenue durant votre inscription :('}
|
|
|
|
def sync_group(self,email,groups):
|
|
db_groups=self.get_group(email)
|
|
json_group=json.dumps(groups)
|
|
if db_groups!=False:
|
|
log.debug('Database groups : %s' % db_groups)
|
|
log.debug('Provided groups : %s' % groups)
|
|
if 'groups' not in db_groups or db_groups['groups']=={}:
|
|
log.debug('Database group is empty')
|
|
if 'groups' not in groups or groups['groups']=={}:
|
|
log.debug('Database and provided group are empty. Return empty')
|
|
return {'groups': {'groups': {}}}
|
|
else:
|
|
log.debug('Insert provided groups in database and return it')
|
|
if self.do_sql("INSERT INTO groups (email,groups) VALUES ('%s','%s')" % (email,json_group)):
|
|
return {'groups': groups}
|
|
elif 'groups' not in groups or groups['groups']=={}:
|
|
log.debug('Provide group is empty. Return database groups')
|
|
return {'groups': db_groups}
|
|
else:
|
|
log.debug('Update database with provided group and return it')
|
|
if self.do_sql("UPDATE groups SET groups='%s' WHERE email='%s'" % (json_group,email)):
|
|
return {'groups': groups}
|
|
return {'syncerror': 'Erreur inconnu'}
|
|
|
|
def get_group(self,email):
|
|
ret=self.select("SELECT groups FROM groups WHERE email='%s'" % email)
|
|
if ret!=False:
|
|
if len(ret)==1:
|
|
return json.loads(ret[0][0])
|
|
else:
|
|
return {'groups': {}}
|
|
else:
|
|
return False
|