def addComment(self, user, message, idWallpaper):
"""
Ajout d'un comment
"""
try:
col = Bdd().connexion("comment")
col.insert({"message": message, "user": user, "idWallpaper": idWallpaper, "date":datetime.today(), "isReported": False, "notification": None})
return 1000
except:
return 1500 # faudra faire une bonne gestion d'erreurs avec des logs
def addCategory(self, name, description):
"""
"""
try:
col = Bdd().connexion("category")
except:
return 1500
if self.categoryExist(name):
return 1520 # la catégorie existe déjà
else:
col.insert({"name": name, "description": description})
return 1000 # ok
def addUser(self, username, email, passHash):
"""
Ajout d'un utilisateur
"""
print("*******************************Ajout d'un user : "******"user")
col.insert({"username": username, "email": email, "passHash": passHash, "dateInscription": datetime.datetime.utcnow()})
print("*******************************Success ajout d'un user : "******"*******************************Fail ajout d'un user : "+ username)
return 1500 # faudra faire une bonne gestion d'erreurs avec des logs
def addWallpaper(self, file, name, username):
"""
Fonction appellée pour ajouter un wallpaper, retourne un nombre à utiliser pour le path ou enregistrer le wallpaper
"""
try:
col = Bdd().connexion("wallpaper")
except:
return 1500
extension = os.path.splitext(file.name)[1]
#if extension != ("jpg" or "JPG" or "png" or "PNG" or "JPEG" or "bmp" or "BMP" or "gif" or "GIF" or "svg" or "SVG"):
#print("L'extention n'est pas une image")
#return 1500
ID = col.insert({"name": name, "nbVote": 0, "visible": True, "username": username})
ID = str(ID)
fd = open('%s/%s/%s.%s' % (MEDIA_ROOT, "wallpapers", ID, extension), 'wb')
fd.write(file.read())
fd.close()
print("wallpaper extension :" + str(extension))
BASEURL = 'http://127.0.0.1:8000/wallpapers/'
path = '%s%s%s' % (BASEURL, ID, extension)
col.update({"_id": ObjectId(ID)}, {"$set": {"path": path}})
print("id wallpaper: " + str(ID))
result = json.loads(dumps(col.find_one({"_id": ObjectId(ID)}))) # on dump le curseur et on converti en json
result["_id"] = str(result["_id"]["$oid"]) # On converti le ObjectId ($oid) en string
return result
