diff --git a/main.py b/main.py index 0a70e66..d04a222 100644 --- a/main.py +++ b/main.py @@ -1,5 +1,3 @@ -import json - from flask import Flask, request from processor import Processor diff --git a/processor.py b/processor.py index c87f566..0d4dd41 100644 --- a/processor.py +++ b/processor.py @@ -15,7 +15,7 @@ class Processor: def next(self): count_docs = mongo.jokes_collection.count_documents({}) - rnd = random.randrange(count_docs) + rnd = random.randrange(1, count_docs + 1) anek = mongo.jokes_collection.find_one({"id": rnd}) return { "text": anek['text']