|
|
@@ -5,7 +5,7 @@ class db():
|
|
|
def __init__(self):
|
|
|
self.client = MongoClient('localhost', 27017)
|
|
|
self.c_sources=self.client.alexis["sources"]
|
|
|
- self.articles=self.client.alexis["articles"]
|
|
|
+ self.c_articles=self.client.alexis["articles"]
|
|
|
|
|
|
def jsonable(self, el):
|
|
|
if el is None or "_id" not in el:
|
|
|
@@ -14,6 +14,14 @@ class db():
|
|
|
el["_id"]=str(el["_id"])
|
|
|
return el
|
|
|
|
|
|
+
|
|
|
+ def articles(self, id=None):
|
|
|
+ if id is None:
|
|
|
+ return [ self.jsonable(p) for p in self.c_articles.find() ]
|
|
|
+
|
|
|
+ ret=self.c_articles.find_one({"_id":ObjectId(id)})
|
|
|
+ return self.jsonable(ret)
|
|
|
+
|
|
|
def sources(self,id=None):
|
|
|
if id is None:
|
|
|
return [ self.jsonable(p) for p in self.c_sources.find() ]
|
|
|
@@ -27,14 +35,14 @@ class db():
|
|
|
|
|
|
def insert_article(self,o):
|
|
|
try:
|
|
|
- self.articles.insert_one(o)
|
|
|
+ self.c_articles.insert_one(o)
|
|
|
return True
|
|
|
except errors.DuplicateKeyError:
|
|
|
print("Dup")
|
|
|
return False
|
|
|
|
|
|
def article_exists(self,url):
|
|
|
- ret=self.articles.find_one({'url':url})
|
|
|
+ ret=self.c_articles.find_one({'url':url})
|
|
|
if ret is None:
|
|
|
return False
|
|
|
return True
|