Implement basic Question object for querying.

Author: Keiron Pizzey

.gitignore

@@ -128,4 +128,5 @@ crashlytics-build.properties
 
 ### Virtualenv
 
-venv/*
+venv/*
+dev.py

kesh/__init__.py

@@ -1 +1 @@
-# Empty __init__.py
+__version__ = '0.1.dev'

kesh/api/__init__.py

@@ -1 +1 @@
-from .core import Connection
+from .connection import MongoConnection

kesh/api/connection.py

@@ -0,0 +1,60 @@
+from pymongo import MongoClient
+
+
+class Connection(object):
+
+    def __init__(self):
+        super().__init__()
+
+
+class MongoConnection(Connection):
+    """Connection object for connecting to the mongodb database and retrieving data."""
+
+    def __init__(self, db, mongo_options={}):
+        super().__init__()
+
+        self.client = MongoClient(**mongo_options)
+        self.db = self.client[db]
+
+
+    def query(self, d):
+
+        coll_name = d.pop('collection', None)
+        if coll_name is None:
+            raise Exception('Collection param not found in query.')
+
+        coll = self.db[coll_name]
+
+        if 'id' in d:
+            return coll.find_one(d)
+
+        if 'ids' in d.keys():
+            return list(coll.find({'id':{'$in':d['ids']}}))
+
+        if 'all' in d and d['all']:
+            return coll.find()
+
+
+    def __enter__(self):
+        """For use with the "with" statement. Will create an open db connection.
+
+        :return: Client connection.
+        """
+
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """For use with the "with" statement. Will disconnect from db connection.
+        :param exc_type:
+        :param exc_val:
+        :param exc_tb:
+        :return:
+        """
+
+        self.client.close()
+
+
+if __name__ == '__main__':
+
+    with MongoConnection() as conn:
+        print(conn)

kesh/api/core.py

@@ -0,0 +1,8 @@
+
+
+class Post(object):
+
+    def __init__(self, db):
+        super().__init__()
+
+        self.db = db

kesh/api/post.py

@@ -0,0 +1,25 @@
+from .post import Post
+
+class Question(Post):
+
+    def __init__(self, db):
+        super().__init__(db=db)
+
+
+    def get_question_by_id(self, id):
+        d = {'collection':'questions', 'id':id}
+
+        result = self.db.query(d)
+        return result
+
+    def get_question_by_ids(self, ids):
+        d = {'collection':'questions', 'ids':ids}
+
+        result = self.db.query(d)
+        return result
+
+    def get_all_questions(self):
+        d = {'collection':'questions', 'all':True}
+
+        result = self.db.query(d)
+        return result

kesh/api/question.py

@@ -0,0 +1,51 @@
+import re
+import os
+
+from setuptools import setup, find_packages
+
+try:
+    with open('requirements.txt') as f:
+        requirements = f.read().splitlines()
+except FileNotFoundError:
+    requirements = []
+
+try:
+    with open(os.path.join(os.path.dirname(__file__), 'kesh', '__init__.py')) as f:
+        version = re.search(r"__version__ = '(.*)'", f.read()).group(1)
+except FileNotFoundError:
+    version = 'test'
+
+classifiers = [
+    'Development Status :: 1 - Planning',
+
+    'Operating System :: OS Independent',
+    'Operating System :: POSIX',
+    'Operating System :: MacOS :: MacOS X',
+    'Operating System :: Microsoft :: Windows',
+
+    'Programming Language :: Python',
+    'Programming Language :: Python :: 3',
+    'Programming Language :: Python :: 3.3',
+    'Programming Language :: Python :: 3.4',
+    'Programming Language :: Python :: 3 :: Only',
+
+    'Intended Audience :: Developers',
+    'Intended Audience :: Science/Research',
+
+    'Topic :: Internet',
+    'Topic :: Scientific/Engineering :: Information Analysis',
+    'Topic :: Software Development :: Libraries :: Python Modules',
+]
+
+setup(
+    name='kesh',
+    packages=find_packages(),
+    url='http://sopython.com',
+    license='BSD',
+    author='Keiron Pizzey',
+    author_email='[email protected]',
+    description='Database program for accessing Stack Overflow information.',
+    install_requires=requirements,
+    classifiers=classifiers,
+    version=version
+)