def delete_files_for_user(cls, user):
mongodbmanager.connect()
files = cls.find_files_for_user(user)
if files is not None:
count = files.count()
files.delete()
return count
def create_file(cls, name, hash, user):
mongodbmanager.connect()
file = cls.find_file_by_hash_and_sessionid(hash, user.session_id)
if file is None:
# Create new file
newFile = File(name = name, hash = hash, session_id = user.session_id)
newFile.save()
else:
# Update file name
cls.update_file(file, name)
def find_files_by_query(cls, query):
mongodbmanager.connect()
files = File.objects(name__icontains = query)
#do manual distinct
returning_files = []
returning_files_hashes = []
for f in files:
if f.hash not in returning_files_hashes:
returning_files.append(f)
returning_files_hashes.append(f.hash)
return returning_files
def delete_user(cls, user):
mongodbmanager.connect()
user.delete()
def create_user(cls, ip, port):
mongodbmanager.connect()
new_user = User(ip=str(ip), port=port, session_id=hashing.generate_session_id())
new_user.save()
return new_user
def find_user_by_ip_and_port(cls, ip, port):
mongodbmanager.connect()
return User.objects(ip = ip, port = port).first()
def find_user_by_ip(cls, ip):
mongodbmanager.connect()
return User.objects(ip = ip).first()
def delete_file(cls, file):
mongodbmanager.connect()
file.delete()
def update_file(cls, file, new_name):
mongodbmanager.connect()
file.name = new_name
file.save()
def find_files_for_user(cls, user):
mongodbmanager.connect()
return File.objects(session_id = user.session_id)
def count_files_by_hash(cls, hash):
mongodbmanager.connect()
return File.objects(hash = hash).count()
def find_files_by_hash(cls, hash):
mongodbmanager.connect()
return File.objects(hash = hash).all()
def find_file_by_hash_and_sessionid(cls, hash, session_id):
mongodbmanager.connect()
return File.objects(hash = hash, session_id = session_id).first()
def increase_download_count_for_file(cls, file):
mongodbmanager.connect()
file.download_count += 1
file.save()
def delete_all(cls):
mongodbmanager.connect()
User.drop_collection()
def find_user_by_session_id(cls, session_id):
mongodbmanager.connect()
return User.objects(session_id=session_id).first()
def find_file(cls, name, hash, user):
mongodbmanager.connect()
return File.objects(name = name, hash = hash, session_id = user.session_id).first()
