21 lines
372 B
Python
21 lines
372 B
Python
# Searches database popularity of a problem
|
|
|
|
import csv
|
|
|
|
from cs50 import SQL
|
|
|
|
# Open database
|
|
db = SQL("sqlite:///favorites.db")
|
|
|
|
# Prompt user for favorite
|
|
favorite = input("Favorite: ")
|
|
|
|
# Search for title
|
|
rows = db.execute("SELECT COUNT(*) AS n FROM favorites WHERE problem = ?", favorite)
|
|
|
|
# Get first (and only) row
|
|
row = rows[0]
|
|
|
|
# Print popularity
|
|
print(row["n"])
|