def deleteProject(cls, id): project = Database.find_one(collection='projects', query={'_id': id}) if project != None: # edit if element exists Vulnerability.deleteVulnerabilitiesOfProject(report_id=id) Database.delete_one(collection='projects', query={"_id": id}) return True else: return False
def deleteVulnerability(cls, id): vuln = Database.find_one(collection='vulnerabilities', query={'_id': id}) if vuln != None: # edit if element exists Database.delete_one(collection='vulnerabilities', query={"_id": id}) return True else: return False