changeset 0:cfcfa093e4b4

initial commit
author Jeff Hammel <jhammel@mozilla.com>
date Sun, 09 Dec 2012 10:21:43 -0800
parents
children 5abe00d24a2f
files INSTALL.py README.txt setup.py tests/doctest.txt tests/test.py tests/unit.py wsgraph/__init__.py wsgraph/main.py wsgraph/model.py wsgraph/web.py
diffstat 10 files changed, 376 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/INSTALL.py	Sun Dec 09 10:21:43 2012 -0800
@@ -0,0 +1,67 @@
+#!/usr/bin/env python
+
+"""
+installation script for WSGraph
+WSGI graph server
+"""
+
+import os
+import sys
+import urllib2
+import subprocess
+try:
+    from subprocess import check_call as call
+except:
+    from subprocess import call
+
+REPO='http://k0s.org/hg/WSGraph'
+DEST='WSGraph' # name of the virtualenv
+VIRTUALENV='https://raw.github.com/pypa/virtualenv/develop/virtualenv.py'
+
+def which(binary, path=os.environ['PATH']):
+    dirs = path.split(os.pathsep)
+    for dir in dirs:
+        if os.path.isfile(os.path.join(dir, fileName)):
+            return os.path.join(dir, fileName)
+        if os.path.isfile(os.path.join(dir, fileName + ".exe")):
+            return os.path.join(dir, fileName + ".exe")
+
+def main(args=sys.argv[1:]):
+
+    # create a virtualenv
+    virtualenv = which('virtualenv') or which('virtualenv.py')
+    if virtualenv:
+        call([virtualenv, DEST])
+    else:
+        process = subproces.Popen([sys.executable, '-', DEST], stdin=subprocess.PIPE)
+        process.communicate(stdin=urllib2.urlopen(VIRTUALENV).read())
+
+    # create a src directory
+    src = os.path.join(DEST, 'src')
+    os.mkdir(src)
+
+    # clone the repository
+    call(['hg', 'clone', REPO], cwd=src)
+
+    # find the virtualenv python
+    python = None
+    for path in (('bin', 'python'), ('Scripts', 'python.exe')):
+        _python = os.path.join(DEST, *path)
+        if os.path.exists(_python)
+            python = _python
+            break
+    else:
+        raise Exception("Python binary not found in %s" % DEST)
+
+    # find the clone
+    filename = REPO.rstrip('/')
+    filename = filename.split('/')[-1]
+    clone = os.path.join(src, filename)
+    assert os.path.exists(clone), "Clone directory not found in %s" % src
+
+    # ensure setup.py exists
+    assert os.path.exists(os.path.join(clone, 'setup.py')), 'setup.py not found in %s' % clone
+
+    # install the package in develop mode
+    call([python 'setup.py', 'develop'], cwd=clone)
+
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/README.txt	Sun Dec 09 10:21:43 2012 -0800
@@ -0,0 +1,11 @@
+WSGraph
+===========
+
+WSGI graph server
+
+----
+
+Jeff Hammel
+
+http://k0s.org/hg/WSGraph
+
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/setup.py	Sun Dec 09 10:21:43 2012 -0800
@@ -0,0 +1,45 @@
+"""
+setup packaging script for WSGraph
+"""
+
+import os
+
+version = "0.0"
+dependencies = ['MakeItSo', 'webob']
+
+# allow use of setuptools/distribute or distutils
+kw = {}
+try:
+    from setuptools import setup
+    kw['entry_points'] = """
+      [console_scripts]
+      WSGraph = WSGraph.main:main
+      WSGraph-template = WSGraph.template:main
+"""
+    kw['install_requires'] = dependencies
+except ImportError:
+    from distutils.core import setup
+    kw['requires'] = dependencies
+
+try:
+    here = os.path.dirname(os.path.abspath(__file__))
+    description = file(os.path.join(here, 'README.txt')).read()
+except IOError:
+    description = ''
+
+
+setup(name='WSGraph',
+      version=version,
+      description="WSGI graph server",
+      long_description=description,
+      classifiers=[], # Get strings from http://www.python.org/pypi?%3Aaction=list_classifiers
+      author='Jeff Hammel',
+      author_email='jhammel@mozilla.com',
+      url='http://k0s.org/hg/WSGraph',
+      license='',
+      packages=['wsgraph'],
+      include_package_data=True,
+      zip_safe=False,
+      **kw
+      )
+
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/doctest.txt	Sun Dec 09 10:21:43 2012 -0800
@@ -0,0 +1,11 @@
+Test WSGraph
+================
+
+The obligatory imports:
+
+    >>> import wsgraph
+
+Run some tests.  This test will fail, please fix it:
+
+    >>> assert True == False
+
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test.py	Sun Dec 09 10:21:43 2012 -0800
@@ -0,0 +1,61 @@
+#!/usr/bin/env python
+
+"""
+doctest runner
+"""
+
+import doctest
+import os
+import sys
+from optparse import OptionParser
+
+
+def run_tests(raise_on_error=False, report_first=False):
+
+    # add results here
+    results = {}
+
+    # doctest arguments
+    directory = os.path.dirname(os.path.abspath(__file__))
+    extraglobs = {'here': directory}
+    doctest_args = dict(extraglobs=extraglobs, raise_on_error=raise_on_error)
+    doctest_args['optionsflags'] = doctest.ELLIPSIS
+    if report_first:
+        doctest_args['optionflags'] |= doctest.REPORT_ONLY_FIRST_FAILURE
+
+    # gather tests
+    tests =  [test for test in os.listdir(directory)
+              if test.endswith('.txt')]
+
+    # run the tests
+    for test in tests:
+        try:
+            results[test] = doctest.testfile(test, **doctest_args)
+        except doctest.DocTestFailure, failure:
+            raise
+        except doctest.UnexpectedException, failure:
+            raise failure.exc_info[0], failure.exc_info[1], failure.exc_info[2]
+
+    return results
+
+def main(args=sys.argv[1:]):
+
+    # parse command line args
+    parser = OptionParser(description=__doc__)
+    parser.add_option('--raise', dest='raise_on_error',
+                      default=False, action='store_true',
+                      help="raise on first error")
+    parser.add_option('--report-first', dest='report_first',
+                      default=False, action='store_true',
+                      help="report the first error only (all tests will still run)")
+    options, args = parser.parse_args(args)
+
+    # run the tests
+    results = run_tests(**options.__dict__)
+    if sum([i.failed for i in results.values()]):
+        sys.exit(1) # error
+                
+
+if __name__ == '__main__':
+    main()
+
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/unit.py	Sun Dec 09 10:21:43 2012 -0800
@@ -0,0 +1,21 @@
+#!/usr/bin/env python
+
+"""
+unit tests
+"""
+
+import os
+import sys
+import unittest
+
+# globals
+here = os.path.dirname(os.path.abspath(__file__))
+
+class wsgraphUnitTest(unittest.TestCase):
+
+    def test_wsgraph(self):
+        pass
+
+if __name__ == '__main__':
+    unittest.main()
+
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/wsgraph/__init__.py	Sun Dec 09 10:21:43 2012 -0800
@@ -0,0 +1,3 @@
+#
+from main import *
+
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/wsgraph/main.py	Sun Dec 09 10:21:43 2012 -0800
@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+
+"""
+WSGI graph server
+"""
+
+import sys
+import optparse
+
+def main(args=sys.argv[:]):
+
+    # parse command line options
+    usage = '%prog [options]'
+    class PlainDescriptionFormatter(optparse.IndentedHelpFormatter):
+        """description formatter for console script entry point"""
+        def format_description(self, description):
+            if description:
+                return description.strip() + '\n'
+            else:
+                return ''
+    parser = optparse.OptionParser(usage=usage, description=__doc__, formatter=PlainDescriptionFormatter())
+    options, args = parser.parse_args(args)
+
+if __name__ == '__main__':
+  main()
+
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/wsgraph/model.py	Sun Dec 09 10:21:43 2012 -0800
@@ -0,0 +1,54 @@
+from abc import abstractmethod
+
+class GraphModel(object):
+
+    @abstractmethod
+    def node(self, name, **values):
+        """get or set a node"""
+
+    @abstractmethod
+    def nodes(self):
+        """returns a list of all nodes"""
+
+    @abstractmethod
+    def edges(self):
+        """returns a list of all edges"""
+
+    @abstractmethod
+    def edge(self, node1, node2, **values):
+        """returns edge from node1 to node2"""
+
+    def __getitem__(self, key):
+        """
+        if key is a basestring, return the node of that name;
+        if key is a 2-tuple/list, return the edge of that name
+        """
+
+
+class MemoryCache(GraphModel):
+
+    def __init__(self):
+        self._edges = {}
+        self._nodes = {}
+
+    def node(self, name, **values):
+        if values:
+            # setter
+            self._nodes[name] = values
+        else:
+            # getter
+            return self._nodes.get(name, None)
+
+    def nodes(self):
+        return self._nodes.keys()
+
+    def edge(self, node1, node2, **values):
+        if values:
+            # setter
+            self._edges[(node1, node2)] = values
+        else:
+            # getter
+            return self._edges.get((node1, node2), None)
+
+    def edges(self):
+        return self._edges.keys()
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/wsgraph/web.py	Sun Dec 09 10:21:43 2012 -0800
@@ -0,0 +1,77 @@
+#!/usr/bin/env python
+
+"""
+web handler for WSGraph
+
+Formatters are keyed off of:
+- (TODO)
+
+formatters = {0: {
+}
+"""
+
+import json
+from webob import Request, Response, exc
+
+def JSONFormatter(**kwargs):
+    return json.dumps(kwargs, sort_keys=True)
+
+def JSONGraphFormatter(graph):
+    return json.dumps({'nodes': graph.nodes(),
+                       'edges': graph.edges()},
+                      sort_keys=True)
+
+class Dispatcher(object):
+
+    def __init__(self, graph,
+                 graph_formatters=None, # TODO
+                 node_formatters=None, # TODO
+                 edge_formatters=None, # TODO
+                 require_auth=False):
+        self.graph = graph
+        self.require_auth = require_auth
+        self.formatters = {0: JSONGraphFormatter,
+                           1: JSONFormatter,
+                           2: JSONFormatter}
+
+    def __call__(self, environ, start_response):
+        request = Request(environ)
+        import pdb; pdb.set_trace()
+        return response(environ, start_response)
+
+    @staticmethod
+    def path_segments(request):
+        import pdb; pdb.set_trace()
+
+    # HTTP methods
+
+    def GET(self, request):
+        """respond to a GET request"""
+
+        segments = self.path_segments(request)
+
+        # formatter
+        formatter = self.formatters[len(segments)]
+        return Response(content_type='text/plain',
+                        body="WSGraph")
+
+    def POST(self, request):
+        """respond to a POST request"""
+        raise NotImplementedError
+
+    def HEAD(self, request):
+        """respond to a HEAD request"""
+        raise NotImplementedError
+
+if __name__ == '__main__':
+
+    # imports
+    from wsgiref import simple_server
+    from .model import MemoryCache
+
+    graph = MemoryCache()
+
+    app = Dispatcher(graph=graph)
+    server = simple_server.make_server(host='0.0.0.0', port=8080, app=app)
+    server.serve_forever()
+