diff --git a/facebook/classes.py b/facebook/classes.py index 8249d87..a601bac 100644 --- a/facebook/classes.py +++ b/facebook/classes.py @@ -11,7 +11,7 @@ from robobrowser import RoboBrowser from arangodb import db from config import * -from helpers import sleep_, update_cookie +from helpers import sleep_, update_cookie, clean class User: @@ -34,7 +34,7 @@ class User: db.insert_document( self.collection, { - "_key": self.username, + "_key": clean(self.username), "url": self.url, "name": self.name, "profile_pictures": self.profile_pictures, @@ -197,7 +197,7 @@ class Friend: db.insert_document( self.collection, { - "_key": self.username, + "_key": clean(self.username), "url": url_bas + self.url, "name": self.name, }, @@ -215,10 +215,10 @@ class Reaction: self.type = False def get_dict(self): - key = str(self.picture_id) + "_" + str(self.user_name_friend) + key = clean(str(self.picture_id) + "_" + str(self.user_name_friend)) return { - "_to": "members/" + str(self.user), - "_from": "members/" + str(self.user_name_friend), + "_to": "members/" + clean(str(self.user)), + "_from": "members/" + clean(str(self.user_name_friend)), "_key": key, "_id": "picture_reactions/" + key, "picture": self.picture_id, diff --git a/facebook/helpers.py b/facebook/helpers.py index cd9f6b9..9259041 100644 --- a/facebook/helpers.py +++ b/facebook/helpers.py @@ -2,6 +2,7 @@ import pickle import random from datetime import datetime from time import sleep +import re from arangodb import db @@ -71,3 +72,6 @@ def write_error(nr, e="", traceback="", soup="", user="", url="", url_name=""): ) except Exception as e: print(e) + +def clean(string): + return ''.join(re.findall('[A-Za-z0-9-:.@()+,=;$!*\s]+', string)) \ No newline at end of file