def load_data(model, folder='data'):
path = "models/{folder}/{file_name}.json".format(
folder=folder, file_name=pluralize_class(model))
with open(path) as data:
list_data = json.load(data)
for json_data in list_data:
model.save(json_data)
URL = 'localhost'
PORT = 27017
USER_COLLECTION = 'users'
MESSAGE_COLLECTION = 'messages'
TOPIC_COLLECTION = 'topics'
client = MongoClient(URL, PORT)
database = client.bot
UserModel = User(database, USER_COLLECTION)
UserModel.delete_collection() #
MessageModel = Message(database, MESSAGE_COLLECTION)
MessageModel.delete_collection()
TopicModel = Topic(database, TOPIC_COLLECTION)
TopicModel.delete_collection()
load_data(MessageModel)
load_data(TopicModel)
