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={})
