Spaces:
Runtime error
Runtime error
File size: 3,521 Bytes
211843b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
from csv_to_db import ImportCSV
import os
importcsv = ImportCSV("RevisionBankDB")
importcsvqp = ImportCSV("RevisionBankDB",maindb= False)
"""
if __name__ == '__main__':
try:
edexcelpapers = list(importcsv.db.edexcelpapers.find()[:2])
print(edexcelpapers)
#del edexcelpapers["_id"]
#print({"edexcelpaper":edexcelpapers}
except Exception as e:
print({f"error":f"{type(e)},{str(e)}"})
"""
def insert_cards(revision_cards):
try:
data = {"email":"amari.lawal05@gmail.com","revisioncards":revision_cards}
email_exists = importcsv.db.accountrevisioncards.find_one({"email":"amari.lawal05@gmail.com"})
if email_exists: # Checks if email exists
cards_not_exist = []
user_revision_cards = list(importcsv.db.accountrevisioncards.find({"email": "amari.lawal05@gmail.com"}))[0] # Gets the email.
#print(user_revision_cards)
for card in data["revisioncards"]: # Checks if the revision card exists in the database.
if card not in user_revision_cards["revisioncards"]:
cards_not_exist.append(card) # If not, add it to the list.
#cards_that_exist.append(card)
if cards_not_exist != []:
new_cards = cards_not_exist + user_revision_cards["revisioncards"] # adds new cards to the list.
user_revision_cards["revisioncards"] = new_cards # Updates the list.
del user_revision_cards["_id"]
user_revision_cards["email"] = "amari.lawal05@gmail.com" # Sets the email to the current user.
importcsv.db.accountrevisioncards.delete_many({"email":"amari.lawal05@gmail.com"}) # Allows data to be updated.
importcsv.db.accountrevisioncards.insert_one(user_revision_cards) # Inserts the new data.
print({"message":"revision cards updated"})
elif cards_not_exist == []: # If the cards are already in the database, print(a message.
print({"message":"No new cards"})
elif not email_exists:
data["email"] = "amari.lawal05@gmail.com"
importcsv.db.accountrevisioncards.insert_one(data)
print({"message": "revision card stored"})
except Exception as ex:
print(type(ex),ex)
def load_cards(dir, subject):
chapter_revision_cards = []
chapters = os.listdir(dir)
for chapter in chapters:
cards = os.listdir(f"{dir}/{chapter}")
for card in cards:
if card.endswith(".txt"):
with open(f"{dir}/{chapter}/{card}", "r", encoding="utf8") as f:
cardjson = {"subject":f"AS Level {subject.capitalize()}","revisioncardtitle":chapter,"revisioncard":f.read()}
chapter_revision_cards.append(cardjson)
return chapter_revision_cards
if __name__ == "__main__":
physicsdir = f"C:/Users/user1/Desktop/RevisionBank/RevisionBank Scheduler/AS Level Card Sender/physicscards"
computersciencedir = f"C:/Users/user1/Desktop/RevisionBank/RevisionBank Scheduler/AS Level Card Sender/computersciencecards"
if __name__ == "__main__":
#revisioncards = load_cards(physicsdir,"physics")
#insert_cards(revisioncards)
#print(revisioncards[0])
revisioncardscmp = load_cards(computersciencedir,"computer science")
insert_cards(revisioncardscmp)
#print(revisioncardscmp[0])
#load_cards(computersciencedir,"computer science")
|