Brand-spanking new setup.py file.

setup.py

@@ -2,20 +2,130 @@
 #
 #  TODO:
 #   * Figure out how to compile and install documentation automatically
-#   * Add back in installation requirements
 #   * Add download_url
 
+import os
+import subprocess
 
-from distutils.core import setup
+CLASSIFIERS = """\
+Development Status :: 4 - Beta
+Intended Audience :: Science/Research
+Intended Audience :: Developers
+License :: OSI Approved :: BSD License
+Programming Language :: Python
+Topic :: Software Development :: Code Generators
+Topic :: Software Development :: Compilers
+Topic :: Scientific/Engineering :: Mathematics
+Operating System :: Microsoft :: Windows
+Operating System :: POSIX
+Operating System :: Unix
+Operating System :: MacOS
+"""
+NAME                = 'Theano'
+MAINTAINER          = "LISA laboratory, University of Montreal"
+MAINTAINER_EMAIL    = "theano-dev@googlegroups.com"
+DESCRIPTION         = 'Optimizing compiler for mathematical expressions'
+LONG_DESCRIPTION    = ""
+URL                 = "http://deeplearning.net/software/theano/"
+DOWNLOAD_URL        = ""
+LICENSE             = 'BSD'
+CLASSIFIERS         = filter(None, CLASSIFIERS.split('\n'))
+AUTHOR              = "LISA laboratory, University of Montreal"
+AUTHOR_EMAIL        = "thaeno-dev@googlegroups.com"
+PLATFORMS           = ["Windows", "Linux", "Solaris", "Mac OS-X", "Unix"]
+MAJOR               = 0
+MINOR               = 3
+MICRO               = 0
+ISRELEASED          = False
 
-setup(name='Theano',
-            version='hg',
-            description='Optimizing compiler for mathematical expressions',
-            author='LISA laboratory, University of Montreal',
-            author_email='theano-dev@googlegroups.com',
-            url='http://www.deeplearning.net/software/theano',
-            packages=['theano', 'theano.tensor', 'theano.gof',
-                      'theano.compile', 'theano.misc', 'theano.scalar',
-                      'theano.sparse', 
-                      'theano.tensor.nnet', 'theano.tensor.signal'],
-           )
+if MICRO > 0:
+    VERSION             = '%d.%d.%d' % (MAJOR, MINOR, MICRO)
+else:
+    VERSION             = "%d.%d" % (MAJOR, MINOR)
+
+# Return the hg revision as a string -- borrowed from hg_version in NumPy's
+# setup.py file
+def hg_version():
+    def _minimal_ext_cmd(cmd):
+        # construct minimal environment
+        env = {}
+        for k in ['SYSTEMROOT', 'PATH', 'PYTHONPATH']:
+            v = os.environ.get(k)
+            if v is not None:
+                env[k] = v
+        # LANGUAGE is used on win32
+        env['LANGUAGE'] = 'C'
+        env['LANG'] = 'C'
+        env['LC_ALL'] = 'C'
+        out = subprocess.Popen(
+            cmd,
+            stdout = subprocess.PIPE,
+            env=env
+        ).communicate()[0]
+        return out
+    try:
+        out = _minimal_ext_cmd(['hg', '-q', 'id'])
+        HG_REVISION = out.strip().decode('ascii')
+    except OSError:
+        HG_REVISION = "unknown-hg"
+    return HG_REVISION
+
+def write_version_py(filename='theano/version.py'):
+    cnt = """
+# THIS FILE IS GENERATED FROM THEANO SETUP.PY
+short_version = '%(version)s'
+version = '%(version)s'
+hg_revision = '%(hg_revision)s'
+full_version = '%(version)s.dev-%%(hg_revision)s' %% {'hg_revision': hg_revision}
+release = %(isrelease)s
+
+if not release:
+    version = full_version
+"""
+    FULL_VERSION = VERSION
+    if not ISRELEASED:
+        if os.path.exists('.hg'):
+            HG_REVISION = hg_version()
+        elif os.path.exists(filename):
+            # must be a source distribution, use existing version file
+            from theano.version import hg_revision as HG_REVISION
+        else:
+            HG_REVISION = "unknown-hg"
+
+        FULL_VERSION += '.dev-' + HG_REVISION
+
+    a = open(filename, 'w')
+    try:
+        a.write(cnt % {'version': VERSION,
+                       'full_version' : FULL_VERSION,
+                       'hg_revision' : HG_REVISION,
+                       'isrelease': str(ISRELEASED)})
+    except Exception, e:
+        print e
+    finally:
+        a.close()
+
+
+def do_setup():
+    write_version_py()
+    from setuptools import setup, find_packages
+    setup(name=NAME,
+          version=VERSION,
+          description=DESCRIPTION,
+          author=AUTHOR,
+          author_email=AUTHOR_EMAIL,
+          url=URL,
+          license=LICENSE,
+          packages=find_packages(),
+          install_requires=['numpy>=1.3.0', 'scipy>=0.7.0'],
+          package_data={
+              '': ['*.txt', '*.rst', '*.cu', '*.cuh', '*.sh'],
+              'theano.misc': ['*.sh']
+          },
+          keywords=' '.join([
+            'theano', 'math', 'numerical', 'symbolic', 'blas',
+            'numpy', 'gpu', 'autodiff', 'differentiation'
+          ])
+    )
+if __name__ == "__main__":
+    do_setup()