def getProjectNumbersOfVulnerabilities(cls): nos_vulns = [] projects = Database.find(collection='projects', query={}) for project in projects: no_vulns = Database.find(collection='vulnerabilities', query={'report_id': project["_id"]}) if no_vulns != None: nos_vulns.append(no_vulns.count()) return nos_vulns
def getVulnerabilities(cls, report_id): vulns = Database.find(collection="vulnerabilities", query={'report_id': report_id}) critical = [] high = [] medium = [] low = [] info = [] if vulns != None: for vuln in vulns: if vuln["severity"] == "critical": critical.append(vuln) elif vuln["severity"] == "high": high.append(vuln) elif vuln["severity"] == "medium": medium.append(vuln) elif vuln["severity"] == "low": low.append(vuln) elif vuln["severity"] == "info": info.append(vuln) if vulns != None: return critical + high + medium + low + info else: return False
def getfavourites(self): favouritelist = [] favourite = Database.find(collection='favourites', query={'user_id': self._id}) for f in favourite: movie = Movie.Movie.get_by_id(f["movie_id"]) favouritelist.append(movie) return favouritelist
def getliked(self): likedlist = [] liked = Database.find(collection='liked', query={'user_id': self._id}) for l in liked: movie = Movie.Movie.get_by_id(l["movie_id"]) likedlist.append(movie) return likedlist
def getunwatched(self): unwatchedlist = [] unwatched = Database.find(collection='unwatched', query={'user_id': self._id}) for l in unwatched: movie = Movie.Movie.get_by_id(l["movie_id"]) unwatchedlist.append(movie) return unwatchedlist
def getVulnerabilitiesNumber(cls, report_id): vulns = Database.find(collection="vulnerabilities", query={'report_id': report_id}) if vulns != None: return vulns.count() else: return 0
def deleteVulnerabilitiesOfProject(cls, report_id): vuln = Database.find(collection='vulnerabilities', query={'report_id': report_id}) if vuln != None: # edit if element exists print( Database.delete_many(collection='vulnerabilities', query={"report_id": report_id})) return True else: return False
def getVulnerabilitiesSeverities(cls, report_id): critical = high = medium = low = info = 0 vulns = Database.find(collection="vulnerabilities", query={'report_id': report_id}) if vulns != None: for vuln in vulns: if vuln["severity"] == "critical": critical = critical + 1 elif vuln["severity"] == "high": high = high + 1 elif vuln["severity"] == "medium": medium = medium + 1 elif vuln["severity"] == "low": low = low + 1 elif vuln["severity"] == "info": info = info + 1 return [critical, high, medium, low, info] else: return False
def company_login(): if request.method == 'GET': return render_template('login.html') elif request.method == 'POST': company_name = request.form['name'] company_password = request.form['password'] if Company.login_company(company_name, company_password): company = Database.find_one(CompanyConstants.COLLECTION, {"company_name": company_name}) # give the company_id to the session to access later on session["company_id"] = company["_id"] session["company_name"] = company_name Company.update_users(company_name, company["key"], session["company_id"]) users = Database.find(UserConstants.COLLECTION, {"company_id": company["_id"]}) user_names = [] for user_id in users: user_names.append(user_id["user_name"]) return render_template("userPicker.html", user_names=user_names) else: return render_template("home.html", string="please use a valid login")
def all(cls): return [ cls(**elem) for elem in Database.find(ItemConstants.COLLECTION, {}) ]
def all_from_category(category): return Database.find(ItemConstants.COLLECTION, {"category": category})
def get_topmovies(cls): movielist = [] movies = Database.find("movies", {'category': "topmovies"}) for m in movies: movielist.append(cls(**m)) return movielist
def get_unplaced_tasks_anyone(company_id): return Database.find(TaskConstants.COLLECTION, { "company_id": company_id, "responsible_party_id": 0 })
def get_placed_user_tasks(company_id, user_id): return Database.find(TaskConstants.placed_COLLECTION, { "responsible_party_id": user_id, "company_id": company_id })
def get_by_firstname_and_lastname(firstname, lastname): return Database.find(COLLECTION, { "creator_lastname": lastname, "creator_firstname": firstname })
def fetch_upvotes(brd_id): return [upreact for upreact in Database.find("reactions", {"suggestion_id": brd_id, "upvote": "on"})]
def fetch_reaction(brd_id): return [react for react in Database.find("reactions", {"suggestion_id": brd_id})]
def get_action(cls): movielist = [] movies = Database.find("movies", {'category': "Action"}) for m in movies: movielist.append(cls(**m)) return movielist
def get_comedy(cls): movielist = [] movies = Database.find("movies", {'genre': "Comedy"}) for m in movies: movielist.append(cls(**m)) return movielist
def get_animation(cls): movielist = [] movies = Database.find("movies", {'genre': "Animation"}) for m in movies: movielist.append(cls(**m)) return movielist
def get_bollywood(cls): movielist = [] movies = Database.find("movies", {'category': "bollywood"}) for m in movies: movielist.append(cls(**m)) return movielist
def get_placed_tasks(): return Database.find( "placed_tasks", {} ) # this will be sorted by the responsible party and buissness (soon)
def from_blog(id): return [ post for post in Database.find(collection='posts', query={'blog_id': id}) ]
def getProjects(cls): projects = Database.find(collection='projects', query={}) if projects != None: return projects else: return False
def get_tasks(): return Database.find(COLLECTION, {})
def get_users_from_db(company_id): return Database.find(UserConstants.COLLECTION, {"company_id": company_id})
def get_placed_anyone_tasks(company_id, user_id): return Database.find(TaskConstants.placed_COLLECTION, { "placed_by": user_id, "company_id": company_id })
def get_by_name(cls, name): return Database.find(ItemConstants.COLLECTION, {"title": name})
def fetch_downvotes(brd_id): return [downreact for downreact in Database.find("reactions", {"suggestion_id": brd_id, "downvote": "on"})]
def get_all_movies(cls): movielist = [] movies = Database.find("movies", {}) for m in movies: movielist.append(cls(**m)) return movielist
def fetch_flagged_bad(brd_id): return [flagged for flagged in Database.find("reactions", {"suggestion_id": brd_id, "flagged": "on"})]
def fetch_suggestions(): return [suggestions for suggestions in Database.find("suggest", {})]
def find_by_product_id(cls, item_id): return cls(**Database.find(COLLECTION, {"item_id": item_id}))
def fetch_by_id(brd_id): return [suggestion_title for suggestion_title in Database.find("suggest", {"board_id": brd_id})]
def find_by_author_id(cls, author_id): blog_data = Database.find(collection='blogs', query={'author_id': author_id}) return[cls(**blog) for blog in blogs]
def getAll(): return Database.find(collection='sensor', query={})