diff options
| author | Kali Kaneko <kali@leap.se> | 2016-04-18 10:56:20 -0400 | 
|---|---|---|
| committer | Kali Kaneko <kali@leap.se> | 2016-04-18 10:56:20 -0400 | 
| commit | e30e06d9062578e1932b5a6a4c4124a1663e18c2 (patch) | |
| tree | 7c81bef2afd8d32d0179cc0192239271252bc311 | |
| parent | e5796bf55e3db177ee567118519136fd96ada3c4 (diff) | |
| parent | cef15c04610ee188052af78ead8cfe7ea29d81c6 (diff) | |
Merge tag '0.5.1'
Tag leap.bitmask version 0.5.1
# gpg: Signature made Mon 18 Apr 2016 10:52:44 AM BOT
# gpg:                using RSA key 1CAF6C5B9F720808
# gpg: Good signature from "Kaliyuga <kaliyuga@riseup.net>" [ultimate]
# gpg:                 aka "Kali Kaneko (leap communications) <kali@leap.se>" [ultimate]
26 files changed, 2665 insertions, 923 deletions
| diff --git a/CHANGELOG.rst b/CHANGELOG.rst index 25266ab..3c02982 100644 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -4,6 +4,24 @@ Changelog  ---------  ==== +2016 +==== + +0.5.1 Apr 18, 2016 ++++++++++++++++++++ + +Features +~~~~~~~~ +- Add HookableService, allowing inter-service notification for hooks. +- Get events working on windows. +- Optional flag to disable curve authentication. + +Bugfixes +~~~~~~~~ +- `#7536 <https://leap.se/code/issues/7536>`_: zmq authenticator often hangs. + + +====  2015  ==== @@ -20,131 +38,3 @@ Misc  - Bump version to 0.5.0, to correct a versioning mistake in the debian packages.  - Rename extras to 'http' and document dependencies on the README.  - Migrate changelog to rst. - - -0.4.4 Oct 28, 2015 -++++++++++++++++++ -- Consider standalone flag when saving events certificates. Related `#7512 <https://leap.se/code/issues/7512>`_. -- fix wrong ca_cert path inside bundle. -- Workaround for deadlock problem in zmq auth. - -0.4.3 Sep 22, 2015 -++++++++++++++++++ -- Expose async methods for events. Closes: `#7274 <https://leap.se/code/issues/7274>`_. - -0.4.2 Aug 26, 2015 -++++++++++++++++++ -- Add http request timeout. Related to `#7234 <https://leap.se/code/issues/7234>`_. -- Add a flag to disable events framework. Closes:`#7259 <https://leap.se/code/issues/7259>`_ -- Allow passing callback to HTTP client. -- Bugfix: do not add a port string to non-tcp addresses. -- Add close method for http agent. -- Fix code style and tests. -- Bugfix: HTTP timeout was not being cleared on abort. - -0.4.1 Jul 10, 2015 -++++++++++++++++++ -- Fix regexp to allow ipc protocol in zmq sockets. Closes: `#7089 <https://leap.se/code/issues/7089>`_. -- Remove extraneous data from events logs. Closes `#7130 <https://leap.se/code/issues/7130>`_. -- Make https client use Twisted SSL validation and adds a reuse by default behavior on connection pool - -0.4.0 Jun 1, 2015 -+++++++++++++++++ -- Modify leap.common.events to use ZMQ. Closes `#6359 <https://leap.se/code/issues/6359>`_. -- Fix time comparison between local and UTC times that caused the VPN certificates not being correctly downloaded on time. Closes `#6994 <https://leap.se/code/issues/6994>`_. -- Add a HTTPClient the twisted way. - -0.3.10 Jan 26, 2015 -+++++++++++++++++++ -- Consider different possibilities for tmpdir. Related to `#6631 <https://leap.se/code/issues/6631>`_. -- Add support for deferreds to memoize_method decorator -- Extract the environment set up and tear down for tests - -==== -2014 -==== - -0.3.9 Jul 18, 2014 -++++++++++++++++++ -- Include pemfile in the package data. Closes `#5897 <https://leap.se/code/issues/5897>`_. -- Look for bundled cacert.pem in the Resources dir for OSX. - -0.3.8 Jun 6, 2014 -+++++++++++++++++ -- Add Soledad sync status signals. Closes `#5517 <https://leap.se/code/issues/5517>`_. - -0.3.7 Apr 4, 2014 -+++++++++++++++++ -- Add memoized_method decorator. Closes `#4784 <https://leap.se/code/issues/4784>`_. -- Add Soledad invalid auth token event. Closes `#5191 <https://leap.se/code/issues/5191>`_. -- Support str type in email charset detection. - -==== -2013 -==== - -0.3.6 Dec 6, 2013 -+++++++++++++++++ -- Update some documentation and packaging bits. - -0.3.5 Nov 1, 2013 -+++++++++++++++++ -- Move get_email_charset to this module. - -0.3.4 Oct 4, 2013 -+++++++++++++++++ -- Add cert bundle including ca-cert certificate. Closes `#3850 <https://leap.se/code/issues/3850>`_. - -0.3.3 Sep 20, 2013 -++++++++++++++++++ -- Fix events server exception raising when port is occupied by some other process. Closes `#3515 <https://leap.se/code/issues/3515>`_. - -0.3.2 Sep 06, 2013 -++++++++++++++++++ -- Use dirspec instead of plain xdg. Closes `#3574 <https://leap.se/code/issues/3574>`_. -- Correct use of CallbackAlreadyRegistered exception. - -0.3.1 Aug 23, 2013 -++++++++++++++++++ -- Add libssl-dev requirement for pyOpenSSL. -- Make the server ping call be async inside events' ensure_server. Fixes `#3355 <https://leap.se/code/issues/3355>`_. -- Requirements in setup are taken from requirements.pip -- Updated requirements. -- Add IMAP_UNREAD_MAIL event. -- Add events for SMTP relay signaling. Closes `#3464 <https://leap.se/code/issues/3464>`_. -- Add events for imap and keymanager notifications. Closes:`#3480 <https://leap.se/code/issues/3480>`_ -- Add versioneer to handle versioning. - -0.3.0 Aug 9, 2013 -+++++++++++++++++ -- OSX: Fix problem with path prefix not returning the correct value. Fixes `#3273 <https://leap.se/code/issues/3273>`_. -- Check if schema exists before load a config. Related to `#3310 <https://leap.se/code/issues/3310>`_. -- Handle schemas and api versions in base class. Related to `#3310 <https://leap.se/code/issues/3310>`_. - -0.2.7 Jul 26, 2013 -++++++++++++++++++ -- Refactor events so components are now called clients. Closes `#3246 <https://leap.se/code/issues/3246>`_ -- Add leap_check helper method, to use whenever leap_assert does not apply. Related to `#3007 <https://leap.se/code/issues/3007>`_. - -0.2.6 Jul 12, 2013 -++++++++++++++++++ -- Improve leap_assert so that it only prints the traceback from the leap_assert call up. Closes `#2895 <https://leap.se/code/issues/2895>`_ -- Add OSX temp directories to the basetests class. - -0.2.5 Jun 28, 2013 -++++++++++++++++++ -- Bugfix: use the provider's default language as default string. Also take care (and note) a possible case with a problematic provider misconfiguration. Closes `#3029 <https://leap.se/code/issues/3029>`_. -- Add data files to setup and manifest (certificates for tests) -- Allow absolute paths in baseconfig.load -- Fix deprecation warnings -- Fix attempt to fetch private keys from server. -- Fix missing imports -- Add possibility of unregistering callbacks for a signal. -- Add a mechanism for events signaling between components. -- Prioritize the path_extension in the which method so it finds our bundled app before the system one, if any. -- Move the Key Manager to leap client repository. -- Move symmetric encryption code to leap.soledad. -- Refactor opengpg utility functions implementation so it uses a context manager. -- Add OpenPGP sign/verify -- Add RAISE_WINDOW event -- Add AES-256 (CTR mode) encrypting/decrypting functions using PyCrypto. diff --git a/HISTORY.rst b/HISTORY.rst new file mode 100644 index 0000000..3254253 --- /dev/null +++ b/HISTORY.rst @@ -0,0 +1,126 @@ +0.4.4 Oct 28, 2015 +++++++++++++++++++ +- Consider standalone flag when saving events certificates. Related `#7512 <https://leap.se/code/issues/7512>`_. +- fix wrong ca_cert path inside bundle. +- Workaround for deadlock problem in zmq auth. + +0.4.3 Sep 22, 2015 +++++++++++++++++++ +- Expose async methods for events. Closes: `#7274 <https://leap.se/code/issues/7274>`_. + +0.4.2 Aug 26, 2015 +++++++++++++++++++ +- Add http request timeout. Related to `#7234 <https://leap.se/code/issues/7234>`_. +- Add a flag to disable events framework. Closes:`#7259 <https://leap.se/code/issues/7259>`_ +- Allow passing callback to HTTP client. +- Bugfix: do not add a port string to non-tcp addresses. +- Add close method for http agent. +- Fix code style and tests. +- Bugfix: HTTP timeout was not being cleared on abort. + +0.4.1 Jul 10, 2015 +++++++++++++++++++ +- Fix regexp to allow ipc protocol in zmq sockets. Closes: `#7089 <https://leap.se/code/issues/7089>`_. +- Remove extraneous data from events logs. Closes `#7130 <https://leap.se/code/issues/7130>`_. +- Make https client use Twisted SSL validation and adds a reuse by default behavior on connection pool + +0.4.0 Jun 1, 2015 ++++++++++++++++++ +- Modify leap.common.events to use ZMQ. Closes `#6359 <https://leap.se/code/issues/6359>`_. +- Fix time comparison between local and UTC times that caused the VPN certificates not being correctly downloaded on time. Closes `#6994 <https://leap.se/code/issues/6994>`_. +- Add a HTTPClient the twisted way. + +0.3.10 Jan 26, 2015 ++++++++++++++++++++ +- Consider different possibilities for tmpdir. Related to `#6631 <https://leap.se/code/issues/6631>`_. +- Add support for deferreds to memoize_method decorator +- Extract the environment set up and tear down for tests + +==== +2014 +==== + +0.3.9 Jul 18, 2014 +++++++++++++++++++ +- Include pemfile in the package data. Closes `#5897 <https://leap.se/code/issues/5897>`_. +- Look for bundled cacert.pem in the Resources dir for OSX. + +0.3.8 Jun 6, 2014 ++++++++++++++++++ +- Add Soledad sync status signals. Closes `#5517 <https://leap.se/code/issues/5517>`_. + +0.3.7 Apr 4, 2014 ++++++++++++++++++ +- Add memoized_method decorator. Closes `#4784 <https://leap.se/code/issues/4784>`_. +- Add Soledad invalid auth token event. Closes `#5191 <https://leap.se/code/issues/5191>`_. +- Support str type in email charset detection. + +==== +2013 +==== + +0.3.6 Dec 6, 2013 ++++++++++++++++++ +- Update some documentation and packaging bits. + +0.3.5 Nov 1, 2013 ++++++++++++++++++ +- Move get_email_charset to this module. + +0.3.4 Oct 4, 2013 ++++++++++++++++++ +- Add cert bundle including ca-cert certificate. Closes `#3850 <https://leap.se/code/issues/3850>`_. + +0.3.3 Sep 20, 2013 +++++++++++++++++++ +- Fix events server exception raising when port is occupied by some other process. Closes `#3515 <https://leap.se/code/issues/3515>`_. + +0.3.2 Sep 06, 2013 +++++++++++++++++++ +- Use dirspec instead of plain xdg. Closes `#3574 <https://leap.se/code/issues/3574>`_. +- Correct use of CallbackAlreadyRegistered exception. + +0.3.1 Aug 23, 2013 +++++++++++++++++++ +- Add libssl-dev requirement for pyOpenSSL. +- Make the server ping call be async inside events' ensure_server. Fixes `#3355 <https://leap.se/code/issues/3355>`_. +- Requirements in setup are taken from requirements.pip +- Updated requirements. +- Add IMAP_UNREAD_MAIL event. +- Add events for SMTP relay signaling. Closes `#3464 <https://leap.se/code/issues/3464>`_. +- Add events for imap and keymanager notifications. Closes:`#3480 <https://leap.se/code/issues/3480>`_ +- Add versioneer to handle versioning. + +0.3.0 Aug 9, 2013 ++++++++++++++++++ +- OSX: Fix problem with path prefix not returning the correct value. Fixes `#3273 <https://leap.se/code/issues/3273>`_. +- Check if schema exists before load a config. Related to `#3310 <https://leap.se/code/issues/3310>`_. +- Handle schemas and api versions in base class. Related to `#3310 <https://leap.se/code/issues/3310>`_. + +0.2.7 Jul 26, 2013 +++++++++++++++++++ +- Refactor events so components are now called clients. Closes `#3246 <https://leap.se/code/issues/3246>`_ +- Add leap_check helper method, to use whenever leap_assert does not apply. Related to `#3007 <https://leap.se/code/issues/3007>`_. + +0.2.6 Jul 12, 2013 +++++++++++++++++++ +- Improve leap_assert so that it only prints the traceback from the leap_assert call up. Closes `#2895 <https://leap.se/code/issues/2895>`_ +- Add OSX temp directories to the basetests class. + +0.2.5 Jun 28, 2013 +++++++++++++++++++ +- Bugfix: use the provider's default language as default string. Also take care (and note) a possible case with a problematic provider misconfiguration. Closes `#3029 <https://leap.se/code/issues/3029>`_. +- Add data files to setup and manifest (certificates for tests) +- Allow absolute paths in baseconfig.load +- Fix deprecation warnings +- Fix attempt to fetch private keys from server. +- Fix missing imports +- Add possibility of unregistering callbacks for a signal. +- Add a mechanism for events signaling between components. +- Prioritize the path_extension in the which method so it finds our bundled app before the system one, if any. +- Move the Key Manager to leap client repository. +- Move symmetric encryption code to leap.soledad. +- Refactor opengpg utility functions implementation so it uses a context manager. +- Add OpenPGP sign/verify +- Add RAISE_WINDOW event +- Add AES-256 (CTR mode) encrypting/decrypting functions using PyCrypto. diff --git a/MANIFEST.in b/MANIFEST.in index cad7096..92da8db 100644 --- a/MANIFEST.in +++ b/MANIFEST.in @@ -5,3 +5,4 @@ include versioneer.py  include LICENSE  include CHANGELOG  include README.rst +include src/leap/common/_version.py @@ -29,3 +29,17 @@ Using `leap.common.http` needs some extra dependencies (twisted.web >= 14.0.2,  python-service-identity). You can install them by running::    pip install leap.common[http] + + +Running the tests +------------------- +To run the tests, first run the setup with: + +.. code-block:: +pip install -r pkg/requirements.pip +pip install -r pkg/requirements-testing.pip + +After that you can run the tests with + +.. code-block:: +trial leap.common diff --git a/changes/next-changelog.rst b/changes/next-changelog.rst index 9f0b455..1371e2c 100644 --- a/changes/next-changelog.rst +++ b/changes/next-changelog.rst @@ -1,4 +1,4 @@ -0.5.0 +0.5.2  +++++++++++++++++++  Please add lines to this file, they will be moved to the CHANGELOG.rst during diff --git a/pkg/requirements.pip b/pkg/requirements.pip index 02fb189..b2be31f 100644 --- a/pkg/requirements.pip +++ b/pkg/requirements.pip @@ -1,8 +1,7 @@ -jsonschema  #<=0.8 -- are we done with this conflict? -dirspec +jsonschema  pyopenssl  python-dateutil  pyzmq>=14.4.1  txzmq>=0.7.3 - -#autopep8 -- ??? +https://launchpad.net/dirspec/stable-13-10/13.10/+download/dirspec-13.10.tar.gz +-e . @@ -8,3 +8,10 @@ ignore = E731  [flake8]  exclude = versioneer.py,_version.py,*.egg,build,dist,docs  ignore = E731 + +[versioneer] +VCS = git +style = pep440 +versionfile_source = src/leap/common/_version.py +versionfile_build = leap/common/_version.py +tag_prefix =  @@ -20,16 +20,17 @@ setup file for leap.common  import re  from setuptools import setup, find_packages  from setuptools import Command +import versioneer  from pkg import utils -import versioneer -versioneer.versionfile_source = 'src/leap/common/_version.py' -versioneer.versionfile_build = 'leap/common/_version.py' -versioneer.tag_prefix = ''  # tags are like 1.2.0 -versioneer.parentdir_prefix = 'leap.common-' -parsed_reqs = utils.parse_requirements() +requirements = utils.parse_requirements() +dependency_links = [requirement for requirement +                    in requirements if requirement.startswith('http')] +requirements = [requirement for requirement +                in requirements if requirement not in dependency_links] +requirements.append('dirspec')  tests_requirements = [      'mock', @@ -49,11 +50,11 @@ trove_classifiers = [      "Topic :: Utilities"  ] -DOWNLOAD_BASE = ('https://github.com/leapcode/leap_pycommon/' +DOWNLOAD_BASE = ('https://github.com/leapcode/bitmask_client/'                   'archive/%s.tar.gz')  _versions = versioneer.get_versions()  VERSION = _versions['version'] -VERSION_FULL = _versions['full'] +VERSION_REVISION = _versions['full-revisionid']  DOWNLOAD_URL = ""  # get the short version for the download url @@ -62,15 +63,30 @@ if len(_version_short) > 0:      VERSION_SHORT = _version_short[0]      DOWNLOAD_URL = DOWNLOAD_BASE % VERSION_SHORT -cmdclass = versioneer.get_cmdclass() -  class freeze_debianver(Command): +      """      Freezes the version in a debian branch.      To be used after merging the development branch onto the debian one.      """      user_options = [] +    template = r""" +# This file was generated by the `freeze_debianver` command in setup.py +# Using 'versioneer.py' (0.16) from +# revision-control system data, or from the parent directory name of an +# unpacked source archive. Distribution tarballs contain a pre-generated copy +# of this file. + +version_version = '{version}' +full_revisionid = '{full_revisionid}' +""" +    templatefun = r""" + +def get_versions(default={}, verbose=False): +        return {'version': version_version, +                'full-revisionid': full_revisionid} +"""      def initialize_options(self):          pass @@ -84,34 +100,23 @@ class freeze_debianver(Command):          if proceed != "y":              print("He. You scared. Aborting.")              return -        template = r""" -# This file was generated by the `freeze_debianver` command in setup.py -# Using 'versioneer.py' (0.7+) from -# revision-control system data, or from the parent directory name of an -# unpacked source archive. Distribution tarballs contain a pre-generated copy -# of this file. - -version_version = '{version}' -version_full = '{version_full}' -""" -        templatefun = r""" - -def get_versions(default={}, verbose=False): -        return {'version': version_version, 'full': version_full} -""" -        subst_template = template.format( +        subst_template = self.template.format(              version=VERSION_SHORT, -            version_full=VERSION_FULL) + templatefun -        with open(versioneer.versionfile_source, 'w') as f: +            full_revisionid=VERSION_REVISION) + self.templatefun +        versioneer_cfg = versioneer.get_config_from_root('.') +        with open(versioneer_cfg.versionfile_source, 'w') as f:              f.write(subst_template) +  try:      long_description = open('README.rst').read() + '\n\n\n' + \          open('CHANGELOG').read()  except Exception:      long_description = "" +cmdclass = versioneer.get_cmdclass()  cmdclass["freeze_debianver"] = freeze_debianver +  setup(      name='leap.common',      version=VERSION, @@ -134,8 +139,8 @@ setup(      # packages=find_packages('src', exclude=['leap.common.tests']),      packages=find_packages('src'),      test_suite='leap.common.tests', -    install_requires=parsed_reqs, -    # dependency_links=dependency_links, +    install_requires=requirements, +    dependency_links=dependency_links,      tests_require=tests_requirements,      include_package_data=True,      zip_safe=False, diff --git a/src/leap/common/__init__.py b/src/leap/common/__init__.py index 383e198..3b07cf8 100644 --- a/src/leap/common/__init__.py +++ b/src/leap/common/__init__.py @@ -4,7 +4,6 @@ from leap.common import certs  from leap.common import check  from leap.common import files  from leap.common import events -from ._version import get_versions  logger = logging.getLogger(__name__) @@ -17,5 +16,6 @@ except ImportError:  __all__ = ["certs", "check", "files", "events"] +from ._version import get_versions  __version__ = get_versions()['version']  del get_versions diff --git a/src/leap/common/_version.py b/src/leap/common/_version.py index de94ba8..e29d969 100644 --- a/src/leap/common/_version.py +++ b/src/leap/common/_version.py @@ -1,73 +1,157 @@ +  # This file helps to compute a version number in source trees obtained from  # git-archive tarball (such as those provided by githubs download-from-tag -# feature). Distribution tarballs (build by setup.py sdist) and build +# feature). Distribution tarballs (built by setup.py sdist) and build  # directories (produced by setup.py build) will contain a much shorter file  # that just contains the computed version number.  # This file is released into the public domain. Generated by -# versioneer-0.7+ (https://github.com/warner/python-versioneer) +# versioneer-0.16 (https://github.com/warner/python-versioneer) -# these strings will be replaced by git during git-archive +"""Git implementation of _version.py.""" +import errno +import os +import re  import subprocess  import sys -import re -import os.path -IN_LONG_VERSION_PY = True -git_refnames = "$Format:%d$" -git_full = "$Format:%H$" +def get_keywords(): +    """Get the keywords needed to look up the version information.""" +    # these strings will be replaced by git during git-archive. +    # setup.py/versioneer.py will grep for the variable names, so they must +    # each be defined on a line of their own. _version.py will just call +    # get_keywords(). +    git_refnames = "$Format:%d$" +    git_full = "$Format:%H$" +    keywords = {"refnames": git_refnames, "full": git_full} +    return keywords -def run_command(args, cwd=None, verbose=False): -    try: -        # remember shell=False, so use git.cmd on windows, not just git -        p = subprocess.Popen(args, stdout=subprocess.PIPE, cwd=cwd) -    except EnvironmentError: -        e = sys.exc_info()[1] + +class VersioneerConfig: +    """Container for Versioneer configuration parameters.""" + + +def get_config(): +    """Create, populate and return the VersioneerConfig() object.""" +    # these strings are filled in when 'setup.py versioneer' creates +    # _version.py +    cfg = VersioneerConfig() +    cfg.VCS = "git" +    cfg.style = "pep440" +    cfg.tag_prefix = "" +    cfg.parentdir_prefix = "None" +    cfg.versionfile_source = "src/leap/common/_version.py" +    cfg.verbose = False +    return cfg + + +class NotThisMethod(Exception): +    """Exception raised if a method is not valid for the current scenario.""" + + +LONG_VERSION_PY = {} +HANDLERS = {} + + +def register_vcs_handler(vcs, method):  # decorator +    """Decorator to mark a method as the handler for a particular VCS.""" +    def decorate(f): +        """Store f in HANDLERS[vcs][method].""" +        if vcs not in HANDLERS: +            HANDLERS[vcs] = {} +        HANDLERS[vcs][method] = f +        return f +    return decorate + + +def run_command(commands, args, cwd=None, verbose=False, hide_stderr=False): +    """Call the given command(s).""" +    assert isinstance(commands, list) +    p = None +    for c in commands: +        try: +            dispcmd = str([c] + args) +            # remember shell=False, so use git.cmd on windows, not just git +            p = subprocess.Popen([c] + args, cwd=cwd, stdout=subprocess.PIPE, +                                 stderr=(subprocess.PIPE if hide_stderr +                                         else None)) +            break +        except EnvironmentError: +            e = sys.exc_info()[1] +            if e.errno == errno.ENOENT: +                continue +            if verbose: +                print("unable to run %s" % dispcmd) +                print(e) +            return None +    else:          if verbose: -            print("unable to run %s" % args[0]) -            print(e) +            print("unable to find command, tried %s" % (commands,))          return None      stdout = p.communicate()[0].strip() -    if sys.version >= '3': +    if sys.version_info[0] >= 3:          stdout = stdout.decode()      if p.returncode != 0:          if verbose: -            print("unable to run %s (error)" % args[0]) +            print("unable to run %s (error)" % dispcmd)          return None      return stdout -def get_expanded_variables(versionfile_source): +def versions_from_parentdir(parentdir_prefix, root, verbose): +    """Try to determine the version from the parent directory name. + +    Source tarballs conventionally unpack into a directory that includes +    both the project name and a version string. +    """ +    dirname = os.path.basename(root) +    if not dirname.startswith(parentdir_prefix): +        if verbose: +            print("guessing rootdir is '%s', but '%s' doesn't start with " +                  "prefix '%s'" % (root, dirname, parentdir_prefix)) +        raise NotThisMethod("rootdir doesn't start with parentdir_prefix") +    return {"version": dirname[len(parentdir_prefix):], +            "full-revisionid": None, +            "dirty": False, "error": None} + + +@register_vcs_handler("git", "get_keywords") +def git_get_keywords(versionfile_abs): +    """Extract version information from the given file."""      # the code embedded in _version.py can just fetch the value of these -    # variables. When used from setup.py, we don't want to import -    # _version.py, so we do it with a regexp instead. This function is not -    # used from _version.py. -    variables = {} +    # keywords. When used from setup.py, we don't want to import _version.py, +    # so we do it with a regexp instead. This function is not used from +    # _version.py. +    keywords = {}      try: -        f = open(versionfile_source, "r") +        f = open(versionfile_abs, "r")          for line in f.readlines():              if line.strip().startswith("git_refnames ="):                  mo = re.search(r'=\s*"(.*)"', line)                  if mo: -                    variables["refnames"] = mo.group(1) +                    keywords["refnames"] = mo.group(1)              if line.strip().startswith("git_full ="):                  mo = re.search(r'=\s*"(.*)"', line)                  if mo: -                    variables["full"] = mo.group(1) +                    keywords["full"] = mo.group(1)          f.close()      except EnvironmentError:          pass -    return variables +    return keywords -def versions_from_expanded_variables(variables, tag_prefix, verbose=False): -    refnames = variables["refnames"].strip() +@register_vcs_handler("git", "keywords") +def git_versions_from_keywords(keywords, tag_prefix, verbose): +    """Get version information from git keywords.""" +    if not keywords: +        raise NotThisMethod("no keywords at all, weird") +    refnames = keywords["refnames"].strip()      if refnames.startswith("$Format"):          if verbose: -            print("variables are unexpanded, not using") -        return {}  # unexpanded, so not in an unpacked git-archive tarball +            print("keywords are unexpanded, not using") +        raise NotThisMethod("unexpanded keywords, not a git-archive tarball")      refs = set([r.strip() for r in refnames.strip("()").split(",")])      # starting in git-1.8.3, tags are listed as "tag: foo-1.0" instead of      # just "foo-1.0". If we see a "tag: " prefix, prefer those. @@ -83,7 +167,7 @@ def versions_from_expanded_variables(variables, tag_prefix, verbose=False):          # "stabilization", as well as "HEAD" and "master".          tags = set([r for r in refs if re.search(r'\d', r)])          if verbose: -            print("discarding '%s', no digits" % ",".join(refs - tags)) +            print("discarding '%s', no digits" % ",".join(refs-tags))      if verbose:          print("likely tags: %s" % ",".join(sorted(tags)))      for ref in sorted(tags): @@ -93,111 +177,308 @@ def versions_from_expanded_variables(variables, tag_prefix, verbose=False):              if verbose:                  print("picking %s" % r)              return {"version": r, -                    "full": variables["full"].strip()} -    # no suitable tags, so we use the full revision id +                    "full-revisionid": keywords["full"].strip(), +                    "dirty": False, "error": None +                    } +    # no suitable tags, so version is "0+unknown", but full hex is still there      if verbose: -        print("no suitable tags, using full revision id") -    return {"version": variables["full"].strip(), -            "full": variables["full"].strip()} - - -def versions_from_vcs(tag_prefix, versionfile_source, verbose=False): -    # this runs 'git' from the root of the source tree. That either means -    # someone ran a setup.py command (and this code is in versioneer.py, so -    # IN_LONG_VERSION_PY=False, thus the containing directory is the root of -    # the source tree), or someone ran a project-specific entry point (and -    # this code is in _version.py, so IN_LONG_VERSION_PY=True, thus the -    # containing directory is somewhere deeper in the source tree). This only -    # gets called if the git-archive 'subst' variables were *not* expanded, -    # and _version.py hasn't already been rewritten with a short version -    # string, meaning we're inside a checked out source tree. +        print("no suitable tags, using unknown + full revision id") +    return {"version": "0+unknown", +            "full-revisionid": keywords["full"].strip(), +            "dirty": False, "error": "no suitable tags"} -    try: -        here = os.path.abspath(__file__) -    except NameError: -        # some py2exe/bbfreeze/non-CPython implementations don't do __file__ -        return {}  # not always correct - -    # versionfile_source is the relative path from the top of the source tree -    # (where the .git directory might live) to this file. Invert this to find -    # the root from __file__. -    root = here -    if IN_LONG_VERSION_PY: -        for i in range(len(versionfile_source.split("/"))): -            root = os.path.dirname(root) -    else: -        root = os.path.dirname(here) + +@register_vcs_handler("git", "pieces_from_vcs") +def git_pieces_from_vcs(tag_prefix, root, verbose, run_command=run_command): +    """Get version from 'git describe' in the root of the source tree. + +    This only gets called if the git-archive 'subst' keywords were *not* +    expanded, and _version.py hasn't already been rewritten with a short +    version string, meaning we're inside a checked out source tree. +    """      if not os.path.exists(os.path.join(root, ".git")):          if verbose:              print("no .git in %s" % root) -        return {} +        raise NotThisMethod("no .git directory") -    GIT = "git" +    GITS = ["git"]      if sys.platform == "win32": -        GIT = "git.cmd" -    stdout = run_command([GIT, "describe", "--tags", "--dirty", "--always"], -                         cwd=root) -    if stdout is None: -        return {} -    if not stdout.startswith(tag_prefix): -        if verbose: -            print("tag '%s' doesn't start with prefix '%s'" % (stdout, tag_prefix)) -        return {} -    tag = stdout[len(tag_prefix):] -    stdout = run_command([GIT, "rev-parse", "HEAD"], cwd=root) -    if stdout is None: -        return {} -    full = stdout.strip() -    if tag.endswith("-dirty"): -        full += "-dirty" -    return {"version": tag, "full": full} - - -def versions_from_parentdir(parentdir_prefix, versionfile_source, verbose=False): -    if IN_LONG_VERSION_PY: -        # We're running from _version.py. If it's from a source tree -        # (execute-in-place), we can work upwards to find the root of the -        # tree, and then check the parent directory for a version string. If -        # it's in an installed application, there's no hope. -        try: -            here = os.path.abspath(__file__) -        except NameError: -            # py2exe/bbfreeze/non-CPython don't have __file__ -            return {}  # without __file__, we have no hope +        GITS = ["git.cmd", "git.exe"] +    # if there is a tag matching tag_prefix, this yields TAG-NUM-gHEX[-dirty] +    # if there isn't one, this yields HEX[-dirty] (no NUM) +    describe_out = run_command(GITS, ["describe", "--tags", "--dirty", +                                      "--always", "--long", +                                      "--match", "%s*" % tag_prefix], +                               cwd=root) +    # --long was added in git-1.5.5 +    if describe_out is None: +        raise NotThisMethod("'git describe' failed") +    describe_out = describe_out.strip() +    full_out = run_command(GITS, ["rev-parse", "HEAD"], cwd=root) +    if full_out is None: +        raise NotThisMethod("'git rev-parse' failed") +    full_out = full_out.strip() + +    pieces = {} +    pieces["long"] = full_out +    pieces["short"] = full_out[:7]  # maybe improved later +    pieces["error"] = None + +    # parse describe_out. It will be like TAG-NUM-gHEX[-dirty] or HEX[-dirty] +    # TAG might have hyphens. +    git_describe = describe_out + +    # look for -dirty suffix +    dirty = git_describe.endswith("-dirty") +    pieces["dirty"] = dirty +    if dirty: +        git_describe = git_describe[:git_describe.rindex("-dirty")] + +    # now we have TAG-NUM-gHEX or HEX + +    if "-" in git_describe: +        # TAG-NUM-gHEX +        mo = re.search(r'^(.+)-(\d+)-g([0-9a-f]+)$', git_describe) +        if not mo: +            # unparseable. Maybe git-describe is misbehaving? +            pieces["error"] = ("unable to parse git-describe output: '%s'" +                               % describe_out) +            return pieces + +        # tag +        full_tag = mo.group(1) +        if not full_tag.startswith(tag_prefix): +            if verbose: +                fmt = "tag '%s' doesn't start with prefix '%s'" +                print(fmt % (full_tag, tag_prefix)) +            pieces["error"] = ("tag '%s' doesn't start with prefix '%s'" +                               % (full_tag, tag_prefix)) +            return pieces +        pieces["closest-tag"] = full_tag[len(tag_prefix):] + +        # distance: number of commits since tag +        pieces["distance"] = int(mo.group(2)) + +        # commit: short hex revision ID +        pieces["short"] = mo.group(3) + +    else: +        # HEX: no tags +        pieces["closest-tag"] = None +        count_out = run_command(GITS, ["rev-list", "HEAD", "--count"], +                                cwd=root) +        pieces["distance"] = int(count_out)  # total number of commits + +    return pieces + + +def plus_or_dot(pieces): +    """Return a + if we don't already have one, else return a .""" +    if "+" in pieces.get("closest-tag", ""): +        return "." +    return "+" + + +def render_pep440(pieces): +    """Build up version string, with post-release "local version identifier". + +    Our goal: TAG[+DISTANCE.gHEX[.dirty]] . Note that if you +    get a tagged build and then dirty it, you'll get TAG+0.gHEX.dirty + +    Exceptions: +    1: no tags. git_describe was just HEX. 0+untagged.DISTANCE.gHEX[.dirty] +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"] or pieces["dirty"]: +            rendered += plus_or_dot(pieces) +            rendered += "%d.g%s" % (pieces["distance"], pieces["short"]) +            if pieces["dirty"]: +                rendered += ".dirty" +    else: +        # exception #1 +        rendered = "0+untagged.%d.g%s" % (pieces["distance"], +                                          pieces["short"]) +        if pieces["dirty"]: +            rendered += ".dirty" +    return rendered + + +def render_pep440_pre(pieces): +    """TAG[.post.devDISTANCE] -- No -dirty. + +    Exceptions: +    1: no tags. 0.post.devDISTANCE +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"]: +            rendered += ".post.dev%d" % pieces["distance"] +    else: +        # exception #1 +        rendered = "0.post.dev%d" % pieces["distance"] +    return rendered + + +def render_pep440_post(pieces): +    """TAG[.postDISTANCE[.dev0]+gHEX] . + +    The ".dev0" means dirty. Note that .dev0 sorts backwards +    (a dirty tree will appear "older" than the corresponding clean one), +    but you shouldn't be releasing software with -dirty anyways. + +    Exceptions: +    1: no tags. 0.postDISTANCE[.dev0] +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"] or pieces["dirty"]: +            rendered += ".post%d" % pieces["distance"] +            if pieces["dirty"]: +                rendered += ".dev0" +            rendered += plus_or_dot(pieces) +            rendered += "g%s" % pieces["short"] +    else: +        # exception #1 +        rendered = "0.post%d" % pieces["distance"] +        if pieces["dirty"]: +            rendered += ".dev0" +        rendered += "+g%s" % pieces["short"] +    return rendered + + +def render_pep440_old(pieces): +    """TAG[.postDISTANCE[.dev0]] . + +    The ".dev0" means dirty. + +    Eexceptions: +    1: no tags. 0.postDISTANCE[.dev0] +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"] or pieces["dirty"]: +            rendered += ".post%d" % pieces["distance"] +            if pieces["dirty"]: +                rendered += ".dev0" +    else: +        # exception #1 +        rendered = "0.post%d" % pieces["distance"] +        if pieces["dirty"]: +            rendered += ".dev0" +    return rendered + + +def render_git_describe(pieces): +    """TAG[-DISTANCE-gHEX][-dirty]. + +    Like 'git describe --tags --dirty --always'. + +    Exceptions: +    1: no tags. HEX[-dirty]  (note: no 'g' prefix) +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"]: +            rendered += "-%d-g%s" % (pieces["distance"], pieces["short"]) +    else: +        # exception #1 +        rendered = pieces["short"] +    if pieces["dirty"]: +        rendered += "-dirty" +    return rendered + + +def render_git_describe_long(pieces): +    """TAG-DISTANCE-gHEX[-dirty]. + +    Like 'git describe --tags --dirty --always -long'. +    The distance/hash is unconditional. + +    Exceptions: +    1: no tags. HEX[-dirty]  (note: no 'g' prefix) +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        rendered += "-%d-g%s" % (pieces["distance"], pieces["short"]) +    else: +        # exception #1 +        rendered = pieces["short"] +    if pieces["dirty"]: +        rendered += "-dirty" +    return rendered + + +def render(pieces, style): +    """Render the given version pieces into the requested style.""" +    if pieces["error"]: +        return {"version": "unknown", +                "full-revisionid": pieces.get("long"), +                "dirty": None, +                "error": pieces["error"]} + +    if not style or style == "default": +        style = "pep440"  # the default + +    if style == "pep440": +        rendered = render_pep440(pieces) +    elif style == "pep440-pre": +        rendered = render_pep440_pre(pieces) +    elif style == "pep440-post": +        rendered = render_pep440_post(pieces) +    elif style == "pep440-old": +        rendered = render_pep440_old(pieces) +    elif style == "git-describe": +        rendered = render_git_describe(pieces) +    elif style == "git-describe-long": +        rendered = render_git_describe_long(pieces) +    else: +        raise ValueError("unknown style '%s'" % style) + +    return {"version": rendered, "full-revisionid": pieces["long"], +            "dirty": pieces["dirty"], "error": None} + + +def get_versions(): +    """Get version information or return default if unable to do so.""" +    # I am in _version.py, which lives at ROOT/VERSIONFILE_SOURCE. If we have +    # __file__, we can work backwards from there to the root. Some +    # py2exe/bbfreeze/non-CPython implementations don't do __file__, in which +    # case we can only use expanded keywords. + +    cfg = get_config() +    verbose = cfg.verbose + +    try: +        return git_versions_from_keywords(get_keywords(), cfg.tag_prefix, +                                          verbose) +    except NotThisMethod: +        pass + +    try: +        root = os.path.realpath(__file__)          # versionfile_source is the relative path from the top of the source -        # tree to _version.py. Invert this to find the root from __file__. -        root = here -        for i in range(len(versionfile_source.split("/"))): +        # tree (where the .git directory might live) to this file. Invert +        # this to find the root from __file__. +        for i in cfg.versionfile_source.split('/'):              root = os.path.dirname(root) -    else: -        # we're running from versioneer.py, which means we're running from -        # the setup.py in a source tree. sys.argv[0] is setup.py in the root. -        here = os.path.abspath(sys.argv[0]) -        root = os.path.dirname(here) +    except NameError: +        return {"version": "0+unknown", "full-revisionid": None, +                "dirty": None, +                "error": "unable to find root of source tree"} -    # Source tarballs conventionally unpack into a directory that includes -    # both the project name and a version string. -    dirname = os.path.basename(root) -    if not dirname.startswith(parentdir_prefix): -        if verbose: -            print("guessing rootdir is '%s', but '%s' doesn't start with prefix '%s'" % -                  (root, dirname, parentdir_prefix)) -        return None -    return {"version": dirname[len(parentdir_prefix):], "full": ""} - -tag_prefix = "" -parentdir_prefix = "leap.common-" -versionfile_source = "src/leap/common/_version.py" - - -def get_versions(default={"version": "unknown", "full": ""}, verbose=False): -    variables = {"refnames": git_refnames, "full": git_full} -    ver = versions_from_expanded_variables(variables, tag_prefix, verbose) -    if not ver: -        ver = versions_from_vcs(tag_prefix, versionfile_source, verbose) -    if not ver: -        ver = versions_from_parentdir(parentdir_prefix, versionfile_source, -                                      verbose) -    if not ver: -        ver = default -    return ver +    try: +        pieces = git_pieces_from_vcs(cfg.tag_prefix, root, verbose) +        return render(pieces, cfg.style) +    except NotThisMethod: +        pass + +    try: +        if cfg.parentdir_prefix: +            return versions_from_parentdir(cfg.parentdir_prefix, root, verbose) +    except NotThisMethod: +        pass + +    return {"version": "0+unknown", "full-revisionid": None, +            "dirty": None, +            "error": "unable to compute version"} diff --git a/src/leap/common/certs.py b/src/leap/common/certs.py index c49015a..95704a6 100644 --- a/src/leap/common/certs.py +++ b/src/leap/common/certs.py @@ -192,8 +192,8 @@ def get_compatible_ssl_context_factory(cert_path=None):          class WebClientContextFactory(ssl.ClientContextFactory):              """ -            A web context factory which ignores the hostname and port and does no -            certificate verification. +            A web context factory which ignores the hostname and port and does +            no certificate verification.              """              def getContext(self, hostname, port):                  return ssl.ClientContextFactory.getContext(self) diff --git a/src/leap/common/events/auth.py b/src/leap/common/events/auth.py new file mode 100644 index 0000000..db217ca --- /dev/null +++ b/src/leap/common/events/auth.py @@ -0,0 +1,100 @@ +# -*- coding: utf-8 -*- +# auth.py +# Copyright (C) 2016 LEAP +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program.  If not, see <http://www.gnu.org/licenses/>. +""" +ZAP authentication, twisted style. +""" +from zmq import PAIR +from zmq.auth.base import Authenticator, VERSION +from txzmq.connection import ZmqConnection +from zmq.utils.strtypes import b, u + +from twisted.python import log + +from txzmq.connection import ZmqEndpoint, ZmqEndpointType + + +class TxAuthenticator(ZmqConnection): + +    """ +    This does not implement the whole ZAP protocol, but the bare minimum that +    we need. +    """ + +    socketType = PAIR +    address = 'inproc://zeromq.zap.01' +    encoding = 'utf-8' + +    def __init__(self, factory, *args, **kw): +        super(TxAuthenticator, self).__init__(factory, *args, **kw) +        self.authenticator = Authenticator(factory.context) +        self.authenticator._send_zap_reply = self._send_zap_reply + +    def start(self): +        endpoint = ZmqEndpoint(ZmqEndpointType.bind, self.address) +        self.addEndpoints([endpoint]) + +    def messageReceived(self, msg): + +        command = msg[0] + +        if command == b'ALLOW': +            addresses = [u(m, self.encoding) for m in msg[1:]] +            try: +                self.authenticator.allow(*addresses) +            except Exception as e: +                log.err("Failed to allow %s", addresses) + +        elif command == b'CURVE': +            domain = u(msg[1], self.encoding) +            location = u(msg[2], self.encoding) +            self.authenticator.configure_curve(domain, location) + +    def _send_zap_reply(self, request_id, status_code, status_text, +                        user_id='user'): +        """ +        Send a ZAP reply to finish the authentication. +        """ +        user_id = user_id if status_code == b'200' else b'' +        if isinstance(user_id, unicode): +            user_id = user_id.encode(self.encoding, 'replace') +        metadata = b''  # not currently used +        reply = [VERSION, request_id, status_code, status_text, +                 user_id, metadata] +        self.send(reply) + +    def shutdown(self): +        if self.factory: +            super(TxAuthenticator, self).shutdown() + + +class TxAuthenticationRequest(ZmqConnection): + +    socketType = PAIR +    address = 'inproc://zeromq.zap.01' +    encoding = 'utf-8' + +    def start(self): +        endpoint = ZmqEndpoint(ZmqEndpointType.connect, self.address) +        self.addEndpoints([endpoint]) + +    def allow(self, *addresses): +        self.send([b'ALLOW'] + [b(a, self.encoding) for a in addresses]) + +    def configure_curve(self, domain='*', location=''): +        domain = b(domain, self.encoding) +        location = b(location, self.encoding) +        self.send([b'CURVE', domain, location]) diff --git a/src/leap/common/events/catalog.py b/src/leap/common/events/catalog.py index 8bddd2c..9a834b2 100644 --- a/src/leap/common/events/catalog.py +++ b/src/leap/common/events/catalog.py @@ -24,49 +24,54 @@ Events catalog.  EVENTS = [      "CLIENT_SESSION_ID",      "CLIENT_UID", -    "IMAP_CLIENT_LOGIN", -    "IMAP_SERVICE_FAILED_TO_START", -    "IMAP_SERVICE_STARTED", -    "IMAP_UNHANDLED_ERROR", -    "KEYMANAGER_DONE_UPLOADING_KEYS", -    "KEYMANAGER_FINISHED_KEY_GENERATION", -    "KEYMANAGER_KEY_FOUND", -    "KEYMANAGER_KEY_NOT_FOUND", -    "KEYMANAGER_LOOKING_FOR_KEY", -    "KEYMANAGER_STARTED_KEY_GENERATION", -    "MAIL_FETCHED_INCOMING", -    "MAIL_MSG_DECRYPTED", -    "MAIL_MSG_DELETED_INCOMING", -    "MAIL_MSG_PROCESSING", -    "MAIL_MSG_SAVED_LOCALLY", -    "MAIL_UNREAD_MESSAGES",      "RAISE_WINDOW", -    "SMTP_CONNECTION_LOST", -    "SMTP_END_ENCRYPT_AND_SIGN", -    "SMTP_END_SIGN", -    "SMTP_RECIPIENT_ACCEPTED_ENCRYPTED", -    "SMTP_RECIPIENT_ACCEPTED_UNENCRYPTED", -    "SMTP_RECIPIENT_REJECTED", -    "SMTP_SEND_MESSAGE_ERROR", -    "SMTP_SEND_MESSAGE_START", -    "SMTP_SEND_MESSAGE_SUCCESS", -    "SMTP_SERVICE_FAILED_TO_START", -    "SMTP_SERVICE_STARTED", -    "SMTP_START_ENCRYPT_AND_SIGN", -    "SMTP_START_SIGN", -    "SOLEDAD_CREATING_KEYS", -    "SOLEDAD_DONE_CREATING_KEYS", -    "SOLEDAD_DONE_DATA_SYNC", -    "SOLEDAD_DONE_DOWNLOADING_KEYS", -    "SOLEDAD_DONE_UPLOADING_KEYS", -    "SOLEDAD_DOWNLOADING_KEYS", -    "SOLEDAD_INVALID_AUTH_TOKEN", -    "SOLEDAD_NEW_DATA_TO_SYNC", -    "SOLEDAD_SYNC_RECEIVE_STATUS", -    "SOLEDAD_SYNC_SEND_STATUS", -    "SOLEDAD_UPLOADING_KEYS",      "UPDATER_DONE_UPDATING",      "UPDATER_NEW_UPDATES", + +    "KEYMANAGER_DONE_UPLOADING_KEYS",  # (address) +    "KEYMANAGER_FINISHED_KEY_GENERATION",  # (address) +    "KEYMANAGER_KEY_FOUND",  # (address) +    "KEYMANAGER_KEY_NOT_FOUND",  # (address) +    "KEYMANAGER_LOOKING_FOR_KEY",  # (address) +    "KEYMANAGER_STARTED_KEY_GENERATION",  # (address) + +    "SOLEDAD_CREATING_KEYS",  # {uuid, userid} +    "SOLEDAD_DONE_CREATING_KEYS",  # {uuid, userid} +    "SOLEDAD_DONE_DATA_SYNC",  # {uuid, userid} +    "SOLEDAD_DONE_DOWNLOADING_KEYS",  # {uuid, userid} +    "SOLEDAD_DONE_UPLOADING_KEYS",  # {uuid, userid} +    "SOLEDAD_DOWNLOADING_KEYS",  # {uuid, userid} +    "SOLEDAD_INVALID_AUTH_TOKEN",  # {uuid, userid} +    "SOLEDAD_SYNC_RECEIVE_STATUS",  # {uuid, userid} +    "SOLEDAD_SYNC_SEND_STATUS",  # {uuid, userid} +    "SOLEDAD_UPLOADING_KEYS",  # {uuid, userid} +    "SOLEDAD_NEW_DATA_TO_SYNC", + +    "MAIL_FETCHED_INCOMING",  # (userid) +    "MAIL_MSG_DECRYPTED",  # (userid) +    "MAIL_MSG_DELETED_INCOMING",  # (userid) +    "MAIL_MSG_PROCESSING",  # (userid) +    "MAIL_MSG_SAVED_LOCALLY",  # (userid) +    "MAIL_UNREAD_MESSAGES",  # (userid, number) + +    "IMAP_SERVICE_STARTED", +    "IMAP_SERVICE_FAILED_TO_START", +    "IMAP_UNHANDLED_ERROR", +    "IMAP_CLIENT_LOGIN",  # (username) + +    "SMTP_SERVICE_STARTED", +    "SMTP_SERVICE_FAILED_TO_START", +    "SMTP_START_ENCRYPT_AND_SIGN",  # (from_addr) +    "SMTP_END_ENCRYPT_AND_SIGN",  # (from_addr) +    "SMTP_START_SIGN",  # (from_addr) +    "SMTP_END_SIGN",  # (from_addr) +    "SMTP_SEND_MESSAGE_START",  # (from_addr) +    "SMTP_SEND_MESSAGE_SUCCESS",  # (from_addr) +    "SMTP_RECIPIENT_ACCEPTED_ENCRYPTED",  # (userid, dest) +    "SMTP_RECIPIENT_ACCEPTED_UNENCRYPTED",  # (userid, dest) +    "SMTP_CONNECTION_LOST",  # (userid, dest) +    "SMTP_RECIPIENT_REJECTED",  # (userid, dest) +    "SMTP_SEND_MESSAGE_ERROR",  # (userid, dest)  ] diff --git a/src/leap/common/events/client.py b/src/leap/common/events/client.py index 60d24bc..78617de 100644 --- a/src/leap/common/events/client.py +++ b/src/leap/common/events/client.py @@ -63,14 +63,18 @@ logger = logging.getLogger(__name__)  _emit_addr = EMIT_ADDR  _reg_addr = REG_ADDR +_factory = None +_enable_curve = True -def configure_client(emit_addr, reg_addr): -    global _emit_addr, _reg_addr +def configure_client(emit_addr, reg_addr, factory=None, enable_curve=True): +    global _emit_addr, _reg_addr, _factory, _enable_curve      logger.debug("Configuring client with addresses: (%s, %s)" %                   (emit_addr, reg_addr))      _emit_addr = emit_addr      _reg_addr = reg_addr +    _factory = factory +    _enable_curve = enable_curve  class EventsClient(object): @@ -103,7 +107,9 @@ class EventsClient(object):          """          with cls._instance_lock:              if cls._instance is None: -                cls._instance = cls(_emit_addr, _reg_addr) +                cls._instance = cls( +                    _emit_addr, _reg_addr, factory=_factory, +                    enable_curve=_enable_curve)          return cls._instance      def register(self, event, callback, uid=None, replace=False): @@ -270,7 +276,7 @@ class EventsClientThread(threading.Thread, EventsClient):      A threaded version of the events client.      """ -    def __init__(self, emit_addr, reg_addr): +    def __init__(self, emit_addr, reg_addr, factory=None, enable_curve=True):          """          Initialize the events client.          """ @@ -281,15 +287,22 @@ class EventsClientThread(threading.Thread, EventsClient):          self._config_prefix = os.path.join(              get_path_prefix(flags.STANDALONE), "leap", "events")          self._loop = None +        self._factory = factory          self._context = None          self._push = None          self._sub = None +        if enable_curve: +            self.use_curve = zmq_has_curve() +        else: +            self.use_curve = False +      def _init_zmq(self):          """          Initialize ZMQ connections.          """          self._loop = EventsIOLoop() +        # we need a new context for each thread          self._context = zmq.Context()          # connect SUB first, otherwise we might miss some event sent from this          # same client @@ -311,7 +324,7 @@ class EventsClientThread(threading.Thread, EventsClient):          logger.debug("Connecting %s to %s." % (socktype, address))          socket = self._context.socket(socktype)          # configure curve authentication -        if zmq_has_curve(): +        if self.use_curve:              public, private = maybe_create_and_get_certificates(                  self._config_prefix, "client")              server_public_file = os.path.join( diff --git a/src/leap/common/events/examples/README.txt b/src/leap/common/events/examples/README.txt new file mode 100644 index 0000000..0bb0df6 --- /dev/null +++ b/src/leap/common/events/examples/README.txt @@ -0,0 +1,49 @@ +How to debug +----------------------------------------- +monitor the events socket: +  sudo ngrep -W byline -d any port 9000 + +launch the server: +  python server.py + +launch the client: +  python client.py + +if zmq is available and enabled, you should see encrypted messages passing by +the socket. + +You should see something like the following: + +#### +T 127.0.0.1:9000 -> 127.0.0.1:33122 [AP] +.......... +## +T 127.0.0.1:33122 -> 127.0.0.1:9000 [AP] +........... +## +T 127.0.0.1:9000 -> 127.0.0.1:33122 [AP] +..CURVE............................................... +# +T 127.0.0.1:33122 -> 127.0.0.1:9000 [AP] +.CURVE............................................... +# +T 127.0.0.1:33122 -> 127.0.0.1:9000 [AP] +...HELLO.............................................................................:....^...".....'.S...n......Y...................O.7.+.D.q".*..R...j.....8..qu..~......Ck.G\....:...m....Tg.s..M..x<.. +## +T 127.0.0.1:9000 -> 127.0.0.1:33122 [AP] +...WELCOME..%.'.,Td... I..}...........`..Nm......./_.Je...4.....-.....f<v.|.".jJ...^.D...$lJ..U......g..../w.......\..W.....!........i.v....0...........3..a.5}.@F..v./..$ +# +T 127.0.0.1:33122 -> 127.0.0.1:9000 [AP] +..........INITIATE......!.*.=0.-......D..]{...A\.tz...!2.....A./ +6.......Y.h.N....cb.U.|..f..)....W..3..X.2U.3PGl.........m..95.(......NJ....5.'..W.GQ..B/.....\%.,Q..r.'L5.......{.W<=._.$.(6j.G... +...37.H..Th...'.........0 ........,..q....U..G..M.`!_..w....f.".......... +.d.K.Y.>f.n.kV. +# +T 127.0.0.1:9000 -> 127.0.0.1:33122 [AP] +.2.READY............A...e.)......*.8y....k.<.N1Z.4.. +# +T 127.0.0.1:33122 -> 127.0.0.1:9000 [AP] +.+.MESSAGE........o...*M..,.... +.r..w..[.GwcU +### + diff --git a/src/leap/common/events/examples/client.py b/src/leap/common/events/examples/client.py new file mode 100644 index 0000000..d6d8985 --- /dev/null +++ b/src/leap/common/events/examples/client.py @@ -0,0 +1,2 @@ +from leap.common.events.txclient import emit +emit('stuff!') diff --git a/src/leap/common/events/examples/server.py b/src/leap/common/events/examples/server.py new file mode 100644 index 0000000..f40f8dc --- /dev/null +++ b/src/leap/common/events/examples/server.py @@ -0,0 +1,4 @@ +from twisted.internet import reactor +from leap.common.events.server import ensure_server +reactor.callWhenRunning(ensure_server) +reactor.run() diff --git a/src/leap/common/events/server.py b/src/leap/common/events/server.py index a69202e..05fc23e 100644 --- a/src/leap/common/events/server.py +++ b/src/leap/common/events/server.py @@ -14,33 +14,31 @@  #  # You should have received a copy of the GNU General Public License  # along with this program. If not, see <http://www.gnu.org/licenses/>. - -  """  The server for the events mechanism.  """ - -  import logging +import platform +  import txzmq  from leap.common.zmq_utils import zmq_has_curve -  from leap.common.events.zmq_components import TxZmqServerComponent -if zmq_has_curve(): +if zmq_has_curve() or platform.system() == "Windows": +    # Windows doesn't have ipc sockets, we need to use always tcp      EMIT_ADDR = "tcp://127.0.0.1:9000"      REG_ADDR = "tcp://127.0.0.1:9001"  else:      EMIT_ADDR = "ipc:///tmp/leap.common.events.socket.0"      REG_ADDR = "ipc:///tmp/leap.common.events.socket.1" -  logger = logging.getLogger(__name__) -def ensure_server(emit_addr=EMIT_ADDR, reg_addr=REG_ADDR): +def ensure_server(emit_addr=EMIT_ADDR, reg_addr=REG_ADDR, path_prefix=None, +                  factory=None, enable_curve=True):      """      Make sure the server is running in the given addresses. @@ -52,7 +50,8 @@ def ensure_server(emit_addr=EMIT_ADDR, reg_addr=REG_ADDR):      :return: an events server instance      :rtype: EventsServer      """ -    _server = EventsServer(emit_addr, reg_addr) +    _server = EventsServer(emit_addr, reg_addr, path_prefix, factory=factory, +                           enable_curve=enable_curve)      return _server @@ -62,7 +61,8 @@ class EventsServer(TxZmqServerComponent):      events in another address.      """ -    def __init__(self, emit_addr, reg_addr): +    def __init__(self, emit_addr, reg_addr, path_prefix=None, factory=None, +                 enable_curve=True):          """          Initialize the events server. @@ -71,7 +71,9 @@ class EventsServer(TxZmqServerComponent):          :param reg_addr: The address to which publish events to clients.          :type reg_addr: str          """ -        TxZmqServerComponent.__init__(self) +        TxZmqServerComponent.__init__(self, path_prefix=path_prefix, +                                      factory=factory, +                                      enable_curve=enable_curve)          # bind PULL and PUB sockets          self._pull, self.pull_port = self._zmq_bind(              txzmq.ZmqPullConnection, emit_addr) diff --git a/src/leap/common/events/tests/test_auth.py b/src/leap/common/events/tests/test_auth.py new file mode 100644 index 0000000..78ffd9f --- /dev/null +++ b/src/leap/common/events/tests/test_auth.py @@ -0,0 +1,64 @@ +# -*- coding: utf-8 -*- +# test_zmq_components.py +# Copyright (C) 2014 LEAP +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program.  If not, see <http://www.gnu.org/licenses/>. +""" +Tests for the auth module. +""" +import os + +from twisted.trial import unittest +from txzmq import ZmqFactory + +from leap.common.events import auth +from leap.common.testing.basetest import BaseLeapTest +from leap.common.zmq_utils import PUBLIC_KEYS_PREFIX +from leap.common.zmq_utils import maybe_create_and_get_certificates + +from txzmq.test import _wait + + +class ZmqAuthTestCase(unittest.TestCase, BaseLeapTest): + +    def setUp(self): +        self.setUpEnv(launch_events_server=False) + +        self.factory = ZmqFactory() +        self._config_prefix = os.path.join(self.tempdir, "leap", "events") + +        self.public, self.secret = maybe_create_and_get_certificates( +            self._config_prefix, 'server') + +        self.authenticator = auth.TxAuthenticator(self.factory) +        self.authenticator.start() +        self.auth_req = auth.TxAuthenticationRequest(self.factory) + +    def tearDown(self): +        self.factory.shutdown() +        self.tearDownEnv() + +    def test_curve_auth(self): +        self.auth_req.start() +        self.auth_req.allow('127.0.0.1') +        public_keys_dir = os.path.join(self._config_prefix, PUBLIC_KEYS_PREFIX) +        self.auth_req.configure_curve(domain="*", location=public_keys_dir) + +        def check(ignored): +            authenticator = self.authenticator.authenticator +            certs = authenticator.certs['*'] +            self.failUnlessEqual(authenticator.whitelist, set([u'127.0.0.1'])) +            self.failUnlessEqual(certs[certs.keys()[0]], True) + +        return _wait(0.1).addCallback(check) diff --git a/src/leap/common/tests/test_events.py b/src/leap/common/events/tests/test_events.py index 2ad097e..d8435c6 100644 --- a/src/leap/common/tests/test_events.py +++ b/src/leap/common/events/tests/test_events.py @@ -14,16 +14,18 @@  #  # You should have received a copy of the GNU General Public License  # along with this program. If not, see <http://www.gnu.org/licenses/>. - - +""" +Tests for the events framework +"""  import os  import logging -import time  from twisted.internet.reactor import callFromThread  from twisted.trial import unittest  from twisted.internet import defer +from txzmq import ZmqFactory +  from leap.common.events import server  from leap.common.events import client  from leap.common.events import flags @@ -40,19 +42,22 @@ class EventsGenericClientTestCase(object):      def setUp(self):          flags.set_events_enabled(True) +        self.factory = ZmqFactory()          self._server = server.ensure_server(              emit_addr="tcp://127.0.0.1:0", -            reg_addr="tcp://127.0.0.1:0") +            reg_addr="tcp://127.0.0.1:0", +            factory=self.factory, +            enable_curve=False) +          self._client.configure_client(              emit_addr="tcp://127.0.0.1:%d" % self._server.pull_port, -            reg_addr="tcp://127.0.0.1:%d" % self._server.pub_port) +            reg_addr="tcp://127.0.0.1:%d" % self._server.pub_port, +            factory=self.factory, enable_curve=False)      def tearDown(self): -        self._client.shutdown() -        self._server.shutdown()          flags.set_events_enabled(False) -        # wait a bit for sockets to close properly -        time.sleep(0.1) +        self.factory.shutdown() +        self._client.instance().reset()      def test_client_register(self):          """ diff --git a/src/leap/common/events/txclient.py b/src/leap/common/events/txclient.py index dfd0533..63f12d7 100644 --- a/src/leap/common/events/txclient.py +++ b/src/leap/common/events/txclient.py @@ -58,16 +58,19 @@ class EventsTxClient(TxZmqClientComponent, EventsClient):      """      def __init__(self, emit_addr=EMIT_ADDR, reg_addr=REG_ADDR, -                 path_prefix=None): +                 path_prefix=None, factory=None, enable_curve=True):          """ -        Initialize the events server. +        Initialize the events client.          """ -        TxZmqClientComponent.__init__(self, path_prefix=path_prefix) +        TxZmqClientComponent.__init__( +            self, path_prefix=path_prefix, factory=factory, +            enable_curve=enable_curve)          EventsClient.__init__(self, emit_addr, reg_addr)          # connect SUB first, otherwise we might miss some event sent from this          # same client          self._sub = self._zmq_connect(txzmq.ZmqSubConnection, reg_addr)          self._sub.gotMessage = self._gotMessage +          self._push = self._zmq_connect(txzmq.ZmqPushConnection, emit_addr)      def _gotMessage(self, msg, tag): @@ -122,7 +125,6 @@ class EventsTxClient(TxZmqClientComponent, EventsClient):          callback(event, *content)      def shutdown(self): -        TxZmqClientComponent.shutdown(self)          EventsClient.shutdown(self) diff --git a/src/leap/common/events/zmq_components.py b/src/leap/common/events/zmq_components.py index 51de02c..c533a74 100644 --- a/src/leap/common/events/zmq_components.py +++ b/src/leap/common/events/zmq_components.py @@ -1,6 +1,6 @@  # -*- coding: utf-8 -*-  # zmq.py -# Copyright (C) 2015 LEAP +# Copyright (C) 2015, 2016 LEAP  #  # This program is free software: you can redistribute it and/or modify  # it under the terms of the GNU General Public License as published by @@ -14,60 +14,63 @@  #  # You should have received a copy of the GNU General Public License  # along with this program. If not, see <http://www.gnu.org/licenses/>. - -  """  The server for the events mechanism.  """ - -  import os  import logging  import txzmq  import re -import time  from abc import ABCMeta -# XXX some distros don't package libsodium, so we have to be prepared for -#     absence of zmq.auth  try:      import zmq.auth -    from zmq.auth.thread import ThreadAuthenticator +    from leap.common.events.auth import TxAuthenticator +    from leap.common.events.auth import TxAuthenticationRequest  except ImportError:      pass +from txzmq.connection import ZmqEndpoint, ZmqEndpointType +  from leap.common.config import flags, get_path_prefix  from leap.common.zmq_utils import zmq_has_curve  from leap.common.zmq_utils import maybe_create_and_get_certificates  from leap.common.zmq_utils import PUBLIC_KEYS_PREFIX -  logger = logging.getLogger(__name__) -  ADDRESS_RE = re.compile("^([a-z]+)://([^:]+):?(\d+)?$") +LOCALHOST_ALLOWED = '127.0.0.1' +  class TxZmqComponent(object):      """      A twisted-powered zmq events component.      """ +    _factory = txzmq.ZmqFactory() +    _factory.registerForShutdown() +    _auth = None      __metaclass__ = ABCMeta      _component_type = None -    def __init__(self, path_prefix=None): +    def __init__(self, path_prefix=None, enable_curve=True, factory=None):          """          Initialize the txzmq component.          """ -        self._factory = txzmq.ZmqFactory() -        self._factory.registerForShutdown()          if path_prefix is None:              path_prefix = get_path_prefix(flags.STANDALONE) +        if factory is not None: +            self._factory = factory          self._config_prefix = os.path.join(path_prefix, "leap", "events")          self._connections = [] +        if enable_curve: +            self.use_curve = zmq_has_curve() +        else: +            self.use_curve = False      @property      def component_type(self): @@ -77,105 +80,89 @@ class TxZmqComponent(object):                  "define a self._component_type!")          return self._component_type -    def _zmq_connect(self, connClass, address): +    def _zmq_bind(self, connClass, address):          """ -        Connect to an address. +        Bind to an address.          :param connClass: The connection class to be used.          :type connClass: txzmq.ZmqConnection -        :param address: The address to connect to. +        :param address: The address to bind to.          :type address: str -        :return: The binded connection. -        :rtype: txzmq.ZmqConnection +        :return: The binded connection and port. +        :rtype: (txzmq.ZmqConnection, int)          """ +        proto, addr, port = ADDRESS_RE.search(address).groups() + +        endpoint = ZmqEndpoint(ZmqEndpointType.bind, address)          connection = connClass(self._factory) -        # create and configure socket -        socket = connection.socket -        if zmq_has_curve(): + +        if self.use_curve: +            socket = connection.socket +              public, secret = maybe_create_and_get_certificates(                  self._config_prefix, self.component_type) -            server_public_file = os.path.join( -                self._config_prefix, PUBLIC_KEYS_PREFIX, "server.key") -            server_public, _ = zmq.auth.load_certificate(server_public_file)              socket.curve_publickey = public              socket.curve_secretkey = secret -            socket.curve_serverkey = server_public -        socket.connect(address) -        logger.debug("Connected %s to %s." % (connClass, address)) -        self._connections.append(connection) -        return connection +            self._start_authentication(connection.socket) -    def _zmq_bind(self, connClass, address): +        if proto == 'tcp' and int(port) == 0: +            connection.endpoints.extend([endpoint]) +            port = connection.socket.bind_to_random_port('tcp://%s' % addr) +        else: +            connection.addEndpoints([endpoint]) + +        return connection, int(port) + +    def _zmq_connect(self, connClass, address):          """ -        Bind to an address. +        Connect to an address.          :param connClass: The connection class to be used.          :type connClass: txzmq.ZmqConnection -        :param address: The address to bind to. +        :param address: The address to connect to.          :type address: str -        :return: The binded connection and port. -        :rtype: (txzmq.ZmqConnection, int) +        :return: The binded connection. +        :rtype: txzmq.ZmqConnection          """ +        endpoint = ZmqEndpoint(ZmqEndpointType.connect, address)          connection = connClass(self._factory) -        socket = connection.socket -        if zmq_has_curve(): + +        if self.use_curve: +            socket = connection.socket              public, secret = maybe_create_and_get_certificates(                  self._config_prefix, self.component_type) +            server_public_file = os.path.join( +                self._config_prefix, PUBLIC_KEYS_PREFIX, "server.key") + +            server_public, _ = zmq.auth.load_certificate(server_public_file)              socket.curve_publickey = public              socket.curve_secretkey = secret -            self._start_thread_auth(connection.socket) +            socket.curve_serverkey = server_public -        proto, addr, port = ADDRESS_RE.search(address).groups() +        connection.addEndpoints([endpoint]) +        return connection -        if proto == "tcp": -            if port is None or port is '0': -                params = proto, addr -                port = socket.bind_to_random_port("%s://%s" % params) -                logger.debug("Binded %s to %s://%s." % ((connClass,) + params)) -            else: -                params = proto, addr, int(port) -                socket.bind("%s://%s:%d" % params) -                logger.debug( -                    "Binded %s to %s://%s:%d." % ((connClass,) + params)) -        else: -            params = proto, addr -            socket.bind("%s://%s" % params) -            logger.debug( -                "Binded %s to %s://%s" % ((connClass,) + params)) -        self._connections.append(connection) -        return connection, port - -    def _start_thread_auth(self, socket): -        """ -        Start the zmq curve thread authenticator. +    def _start_authentication(self, socket): -        :param socket: The socket in which to configure the authenticator. -        :type socket: zmq.Socket -        """ -        authenticator = ThreadAuthenticator(self._factory.context) +        if not TxZmqComponent._auth: +            TxZmqComponent._auth = TxAuthenticator(self._factory) +            TxZmqComponent._auth.start() -        # Temporary fix until we understand what the problem is -        # See https://leap.se/code/issues/7536 -        time.sleep(0.5) +        auth_req = TxAuthenticationRequest(self._factory) +        auth_req.start() +        auth_req.allow(LOCALHOST_ALLOWED) -        authenticator.start() -        # XXX do not hardcode this here. -        authenticator.allow('127.0.0.1')          # tell authenticator to use the certificate in a directory          public_keys_dir = os.path.join(self._config_prefix, PUBLIC_KEYS_PREFIX) -        authenticator.configure_curve(domain="*", location=public_keys_dir) -        socket.curve_server = True  # must come before bind +        auth_req.configure_curve(domain="*", location=public_keys_dir) +        auth_req.shutdown() +        TxZmqComponent._auth.shutdown() -    def shutdown(self): -        """ -        Shutdown the component. -        """ -        logger.debug("Shutting down component %s." % str(self)) -        for conn in self._connections: -            conn.shutdown() -        self._factory.shutdown() +        # This has to be set before binding the socket, that's why this method +        # has to be called before addEndpoints() +        socket.curve_server = True  class TxZmqServerComponent(TxZmqComponent): diff --git a/src/leap/common/service_hooks.py b/src/leap/common/service_hooks.py new file mode 100644 index 0000000..96e95cc --- /dev/null +++ b/src/leap/common/service_hooks.py @@ -0,0 +1,75 @@ +# -*- coding: utf-8 -*- +# service_hooks.py +# Copyright (C) 2016 LEAP +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program.  If not, see <http://www.gnu.org/licenses/>. +""" +Hooks for service composition. +""" +from collections import defaultdict + +from twisted.application.service import IService, Service +from twisted.python import log + +from zope.interface import implementer + + +@implementer(IService) +class HookableService(Service): + +    """ +    This service allows for other services in a Twisted Service tree to be +    notified whenever a certain kind of hook is triggered. + +    During the service composition, one is expected to register +    a hook name with the name of the service that wants to react to the +    triggering of the hook. All the services, both hooked and listeners, should +    be registered against the same parent service. + +    Upon the hook being triggered, the method "hook_<name>" will be called with +    the passed data in the listener service. +    """ + +    def register_hook(self, name, listener): +        if not hasattr(self, 'event_listeners'): +            self.event_listeners = defaultdict(list) +        log.msg("Registering hook %s->%s" % (name, listener)) +        self.event_listeners[name].append(listener) + +    def trigger_hook(self, name, **data): + +        def react_to_hook(listener, name, **kw): +            try: +                getattr(listener, 'hook_' + name)(**kw) +            except AttributeError: +                raise RuntimeError( +                    "Tried to notify a hook, but the listener service class %s" +                    "has not defined the proper method" % listener.__class__) + +        if not hasattr(self, 'event_listeners'): +            self.event_listeners = defaultdict(list) +        listeners = self._get_listener_services(name) + +        for listener in listeners: +            react_to_hook(listener, name, **data) + +    def _get_sibling_service(self, name): +        return self.parent.getServiceNamed(name) + +    def _get_listener_services(self, hook): +        if hook in self.event_listeners: +            service_names = self.event_listeners[hook] +            services = [ +                self._get_sibling_service(name) for name in service_names] +            return services diff --git a/src/leap/common/testing/basetest.py b/src/leap/common/testing/basetest.py index 3d3cee0..2e84a25 100644 --- a/src/leap/common/testing/basetest.py +++ b/src/leap/common/testing/basetest.py @@ -52,7 +52,7 @@ class BaseLeapTest(unittest.TestCase):          cls.tearDownEnv()      @classmethod -    def setUpEnv(cls): +    def setUpEnv(cls, launch_events_server=True):          """          Sets up common facilities for testing this TestCase:          - custom PATH and HOME environmental variables @@ -72,14 +72,15 @@ class BaseLeapTest(unittest.TestCase):          os.environ["PATH"] = bin_tdir          os.environ["HOME"] = cls.tempdir          os.environ["XDG_CONFIG_HOME"] = os.path.join(cls.tempdir, ".config") -        cls._init_events() +        if launch_events_server: +            cls._init_events()      @classmethod      def _init_events(cls):          if flags.EVENTS_ENABLED:              cls._server = events_server.ensure_server( -                emit_addr="tcp://127.0.0.1:0", -                reg_addr="tcp://127.0.0.1:0") +                emit_addr="tcp://127.0.0.1", +                reg_addr="tcp://127.0.0.1")              events_client.configure_client(                  emit_addr="tcp://127.0.0.1:%d" % cls._server.pull_port,                  reg_addr="tcp://127.0.0.1:%d" % cls._server.pub_port) diff --git a/src/leap/common/zmq_utils.py b/src/leap/common/zmq_utils.py index 0a781de..39a49c7 100644 --- a/src/leap/common/zmq_utils.py +++ b/src/leap/common/zmq_utils.py @@ -19,6 +19,7 @@ Utilities to handle ZMQ certificates.  """  import os  import logging +import platform  import stat  import shutil @@ -52,6 +53,10 @@ def zmq_has_curve():         `zmq.auth` module is new in version 14.1         `zmq.has()` is new in version 14.1, new in version libzmq-4.1.      """ +    if platform.system() == "Windows": +        # TODO: curve is not working on windows #7919 +        return False +      zmq_version = zmq.zmq_version_info()      pyzmq_version = zmq.pyzmq_version_info() diff --git a/versioneer.py b/versioneer.py index 4e2c0a5..7ed2a21 100644 --- a/versioneer.py +++ b/versioneer.py @@ -1,170 +1,641 @@ -#! /usr/bin/python -"""versioneer.py +# Version: 0.16 -(like a rocketeer, but for versions) +"""The Versioneer - like a rocketeer, but for versions. +The Versioneer +============== + +* like a rocketeer, but for versions!  * https://github.com/warner/python-versioneer  * Brian Warner  * License: Public Domain -* Version: 0.7+ - -This file helps distutils-based projects manage their version number by just -creating version-control tags. - -For developers who work from a VCS-generated tree (e.g. 'git clone' etc), -each 'setup.py version', 'setup.py build', 'setup.py sdist' will compute a -version number by asking your version-control tool about the current -checkout. The version number will be written into a generated _version.py -file of your choosing, where it can be included by your __init__.py - -For users who work from a VCS-generated tarball (e.g. 'git archive'), it will -compute a version number by looking at the name of the directory created when -te tarball is unpacked. This conventionally includes both the name of the -project and a version number. - -For users who work from a tarball built by 'setup.py sdist', it will get a -version number from a previously-generated _version.py file. - -As a result, loading code directly from the source tree will not result in a -real version. If you want real versions from VCS trees (where you frequently -update from the upstream repository, or do new development), you will need to -do a 'setup.py version' after each update, and load code from the build/ -directory. - -You need to provide this code with a few configuration values: - - versionfile_source: -    A project-relative pathname into which the generated version strings -    should be written. This is usually a _version.py next to your project's -    main __init__.py file. If your project uses src/myproject/__init__.py, -    this should be 'src/myproject/_version.py'. This file should be checked -    in to your VCS as usual: the copy created below by 'setup.py -    update_files' will include code that parses expanded VCS keywords in -    generated tarballs. The 'build' and 'sdist' commands will replace it with -    a copy that has just the calculated version string. - - versionfile_build: -    Like versionfile_source, but relative to the build directory instead of -    the source directory. These will differ when your setup.py uses -    'package_dir='. If you have package_dir={'myproject': 'src/myproject'}, -    then you will probably have versionfile_build='myproject/_version.py' and -    versionfile_source='src/myproject/_version.py'. - - tag_prefix: a string, like 'PROJECTNAME-', which appears at the start of all -             VCS tags. If your tags look like 'myproject-1.2.0', then you -             should use tag_prefix='myproject-'. If you use unprefixed tags -             like '1.2.0', this should be an empty string. - - parentdir_prefix: a string, frequently the same as tag_prefix, which -                   appears at the start of all unpacked tarball filenames. If -                   your tarball unpacks into 'myproject-1.2.0', this should -                   be 'myproject-'. - -To use it: - - 1: include this file in the top level of your project - 2: make the following changes to the top of your setup.py: -     import versioneer -     versioneer.versionfile_source = 'src/myproject/_version.py' -     versioneer.versionfile_build = 'myproject/_version.py' -     versioneer.tag_prefix = '' # tags are like 1.2.0 -     versioneer.parentdir_prefix = 'myproject-' # dirname like 'myproject-1.2.0' - 3: add the following arguments to the setup() call in your setup.py: -     version=versioneer.get_version(), -     cmdclass=versioneer.get_cmdclass(), - 4: run 'setup.py update_files', which will create _version.py, and will -    modify your __init__.py to define __version__ (by calling a function -    from _version.py) - 5: modify your MANIFEST.in to include versioneer.py - 6: add both versioneer.py and the generated _version.py to your VCS -""" +* Compatible With: python2.6, 2.7, 3.3, 3.4, 3.5, and pypy +* [![Latest Version] +(https://pypip.in/version/versioneer/badge.svg?style=flat) +](https://pypi.python.org/pypi/versioneer/) +* [![Build Status] +(https://travis-ci.org/warner/python-versioneer.png?branch=master) +](https://travis-ci.org/warner/python-versioneer) + +This is a tool for managing a recorded version number in distutils-based +python projects. The goal is to remove the tedious and error-prone "update +the embedded version string" step from your release process. Making a new +release should be as easy as recording a new tag in your version-control +system, and maybe making new tarballs. + + +## Quick Install + +* `pip install versioneer` to somewhere to your $PATH +* add a `[versioneer]` section to your setup.cfg (see below) +* run `versioneer install` in your source tree, commit the results + +## Version Identifiers + +Source trees come from a variety of places: + +* a version-control system checkout (mostly used by developers) +* a nightly tarball, produced by build automation +* a snapshot tarball, produced by a web-based VCS browser, like github's +  "tarball from tag" feature +* a release tarball, produced by "setup.py sdist", distributed through PyPI + +Within each source tree, the version identifier (either a string or a number, +this tool is format-agnostic) can come from a variety of places: + +* ask the VCS tool itself, e.g. "git describe" (for checkouts), which knows +  about recent "tags" and an absolute revision-id +* the name of the directory into which the tarball was unpacked +* an expanded VCS keyword ($Id$, etc) +* a `_version.py` created by some earlier build step + +For released software, the version identifier is closely related to a VCS +tag. Some projects use tag names that include more than just the version +string (e.g. "myproject-1.2" instead of just "1.2"), in which case the tool +needs to strip the tag prefix to extract the version identifier. For +unreleased software (between tags), the version identifier should provide +enough information to help developers recreate the same tree, while also +giving them an idea of roughly how old the tree is (after version 1.2, before +version 1.3). Many VCS systems can report a description that captures this, +for example `git describe --tags --dirty --always` reports things like +"0.7-1-g574ab98-dirty" to indicate that the checkout is one revision past the +0.7 tag, has a unique revision id of "574ab98", and is "dirty" (it has +uncommitted changes. + +The version identifier is used for multiple purposes: + +* to allow the module to self-identify its version: `myproject.__version__` +* to choose a name and prefix for a 'setup.py sdist' tarball + +## Theory of Operation + +Versioneer works by adding a special `_version.py` file into your source +tree, where your `__init__.py` can import it. This `_version.py` knows how to +dynamically ask the VCS tool for version information at import time. + +`_version.py` also contains `$Revision$` markers, and the installation +process marks `_version.py` to have this marker rewritten with a tag name +during the `git archive` command. As a result, generated tarballs will +contain enough information to get the proper version. + +To allow `setup.py` to compute a version too, a `versioneer.py` is added to +the top level of your source tree, next to `setup.py` and the `setup.cfg` +that configures it. This overrides several distutils/setuptools commands to +compute the version when invoked, and changes `setup.py build` and `setup.py +sdist` to replace `_version.py` with a small static file that contains just +the generated version data. + +## Installation + +First, decide on values for the following configuration variables: + +* `VCS`: the version control system you use. Currently accepts "git". + +* `style`: the style of version string to be produced. See "Styles" below for +  details. Defaults to "pep440", which looks like +  `TAG[+DISTANCE.gSHORTHASH[.dirty]]`. + +* `versionfile_source`: + +  A project-relative pathname into which the generated version strings should +  be written. This is usually a `_version.py` next to your project's main +  `__init__.py` file, so it can be imported at runtime. If your project uses +  `src/myproject/__init__.py`, this should be `src/myproject/_version.py`. +  This file should be checked in to your VCS as usual: the copy created below +  by `setup.py setup_versioneer` will include code that parses expanded VCS +  keywords in generated tarballs. The 'build' and 'sdist' commands will +  replace it with a copy that has just the calculated version string. + +  This must be set even if your project does not have any modules (and will +  therefore never import `_version.py`), since "setup.py sdist" -based trees +  still need somewhere to record the pre-calculated version strings. Anywhere +  in the source tree should do. If there is a `__init__.py` next to your +  `_version.py`, the `setup.py setup_versioneer` command (described below) +  will append some `__version__`-setting assignments, if they aren't already +  present. + +* `versionfile_build`: + +  Like `versionfile_source`, but relative to the build directory instead of +  the source directory. These will differ when your setup.py uses +  'package_dir='. If you have `package_dir={'myproject': 'src/myproject'}`, +  then you will probably have `versionfile_build='myproject/_version.py'` and +  `versionfile_source='src/myproject/_version.py'`. + +  If this is set to None, then `setup.py build` will not attempt to rewrite +  any `_version.py` in the built tree. If your project does not have any +  libraries (e.g. if it only builds a script), then you should use +  `versionfile_build = None`. To actually use the computed version string, +  your `setup.py` will need to override `distutils.command.build_scripts` +  with a subclass that explicitly inserts a copy of +  `versioneer.get_version()` into your script file. See +  `test/demoapp-script-only/setup.py` for an example. + +* `tag_prefix`: + +  a string, like 'PROJECTNAME-', which appears at the start of all VCS tags. +  If your tags look like 'myproject-1.2.0', then you should use +  tag_prefix='myproject-'. If you use unprefixed tags like '1.2.0', this +  should be an empty string, using either `tag_prefix=` or `tag_prefix=''`. + +* `parentdir_prefix`: + +  a optional string, frequently the same as tag_prefix, which appears at the +  start of all unpacked tarball filenames. If your tarball unpacks into +  'myproject-1.2.0', this should be 'myproject-'. To disable this feature, +  just omit the field from your `setup.cfg`. + +This tool provides one script, named `versioneer`. That script has one mode, +"install", which writes a copy of `versioneer.py` into the current directory +and runs `versioneer.py setup` to finish the installation. + +To versioneer-enable your project: + +* 1: Modify your `setup.cfg`, adding a section named `[versioneer]` and +  populating it with the configuration values you decided earlier (note that +  the option names are not case-sensitive): + +  ```` +  [versioneer] +  VCS = git +  style = pep440 +  versionfile_source = src/myproject/_version.py +  versionfile_build = myproject/_version.py +  tag_prefix = +  parentdir_prefix = myproject- +  ```` + +* 2: Run `versioneer install`. This will do the following: + +  * copy `versioneer.py` into the top of your source tree +  * create `_version.py` in the right place (`versionfile_source`) +  * modify your `__init__.py` (if one exists next to `_version.py`) to define +    `__version__` (by calling a function from `_version.py`) +  * modify your `MANIFEST.in` to include both `versioneer.py` and the +    generated `_version.py` in sdist tarballs + +  `versioneer install` will complain about any problems it finds with your +  `setup.py` or `setup.cfg`. Run it multiple times until you have fixed all +  the problems. + +* 3: add a `import versioneer` to your setup.py, and add the following +  arguments to the setup() call: + +        version=versioneer.get_version(), +        cmdclass=versioneer.get_cmdclass(), + +* 4: commit these changes to your VCS. To make sure you won't forget, +  `versioneer install` will mark everything it touched for addition using +  `git add`. Don't forget to add `setup.py` and `setup.cfg` too. + +## Post-Installation Usage + +Once established, all uses of your tree from a VCS checkout should get the +current version string. All generated tarballs should include an embedded +version string (so users who unpack them will not need a VCS tool installed). + +If you distribute your project through PyPI, then the release process should +boil down to two steps: + +* 1: git tag 1.0 +* 2: python setup.py register sdist upload + +If you distribute it through github (i.e. users use github to generate +tarballs with `git archive`), the process is: + +* 1: git tag 1.0 +* 2: git push; git push --tags + +Versioneer will report "0+untagged.NUMCOMMITS.gHASH" until your tree has at +least one tag in its history. + +## Version-String Flavors + +Code which uses Versioneer can learn about its version string at runtime by +importing `_version` from your main `__init__.py` file and running the +`get_versions()` function. From the "outside" (e.g. in `setup.py`), you can +import the top-level `versioneer.py` and run `get_versions()`. + +Both functions return a dictionary with different flavors of version +information: + +* `['version']`: A condensed version string, rendered using the selected +  style. This is the most commonly used value for the project's version +  string. The default "pep440" style yields strings like `0.11`, +  `0.11+2.g1076c97`, or `0.11+2.g1076c97.dirty`. See the "Styles" section +  below for alternative styles. + +* `['full-revisionid']`: detailed revision identifier. For Git, this is the +  full SHA1 commit id, e.g. "1076c978a8d3cfc70f408fe5974aa6c092c949ac". + +* `['dirty']`: a boolean, True if the tree has uncommitted changes. Note that +  this is only accurate if run in a VCS checkout, otherwise it is likely to +  be False or None + +* `['error']`: if the version string could not be computed, this will be set +  to a string describing the problem, otherwise it will be None. It may be +  useful to throw an exception in setup.py if this is set, to avoid e.g. +  creating tarballs with a version string of "unknown". + +Some variants are more useful than others. Including `full-revisionid` in a +bug report should allow developers to reconstruct the exact code being tested +(or indicate the presence of local changes that should be shared with the +developers). `version` is suitable for display in an "about" box or a CLI +`--version` output: it can be easily compared against release notes and lists +of bugs fixed in various releases. + +The installer adds the following text to your `__init__.py` to place a basic +version in `YOURPROJECT.__version__`: + +    from ._version import get_versions +    __version__ = get_versions()['version'] +    del get_versions + +## Styles + +The setup.cfg `style=` configuration controls how the VCS information is +rendered into a version string. + +The default style, "pep440", produces a PEP440-compliant string, equal to the +un-prefixed tag name for actual releases, and containing an additional "local +version" section with more detail for in-between builds. For Git, this is +TAG[+DISTANCE.gHEX[.dirty]] , using information from `git describe --tags +--dirty --always`. For example "0.11+2.g1076c97.dirty" indicates that the +tree is like the "1076c97" commit but has uncommitted changes (".dirty"), and +that this commit is two revisions ("+2") beyond the "0.11" tag. For released +software (exactly equal to a known tag), the identifier will only contain the +stripped tag, e.g. "0.11". + +Other styles are available. See details.md in the Versioneer source tree for +descriptions. + +## Debugging + +Versioneer tries to avoid fatal errors: if something goes wrong, it will tend +to return a version of "0+unknown". To investigate the problem, run `setup.py +version`, which will run the version-lookup code in a verbose mode, and will +display the full contents of `get_versions()` (including the `error` string, +which may help identify what went wrong). + +## Updating Versioneer + +To upgrade your project to a new release of Versioneer, do the following: + +* install the new Versioneer (`pip install -U versioneer` or equivalent) +* edit `setup.cfg`, if necessary, to include any new configuration settings +  indicated by the release notes +* re-run `versioneer install` in your source tree, to replace +  `SRC/_version.py` +* commit any changed files + +### Upgrading to 0.16 + +Nothing special. + +### Upgrading to 0.15 + +Starting with this version, Versioneer is configured with a `[versioneer]` +section in your `setup.cfg` file. Earlier versions required the `setup.py` to +set attributes on the `versioneer` module immediately after import. The new +version will refuse to run (raising an exception during import) until you +have provided the necessary `setup.cfg` section. + +In addition, the Versioneer package provides an executable named +`versioneer`, and the installation process is driven by running `versioneer +install`. In 0.14 and earlier, the executable was named +`versioneer-installer` and was run without an argument. + +### Upgrading to 0.14 -import os, sys, re -from distutils.core import Command -from distutils.command.sdist import sdist as _sdist -from distutils.command.build import build as _build +0.14 changes the format of the version string. 0.13 and earlier used +hyphen-separated strings like "0.11-2-g1076c97-dirty". 0.14 and beyond use a +plus-separated "local version" section strings, with dot-separated +components, like "0.11+2.g1076c97". PEP440-strict tools did not like the old +format, but should be ok with the new one. -versionfile_source = None -versionfile_build = None -tag_prefix = None -parentdir_prefix = None +### Upgrading from 0.11 to 0.12 -VCS = "git" -IN_LONG_VERSION_PY = False +Nothing special. +### Upgrading from 0.10 to 0.11 -LONG_VERSION_PY = ''' -IN_LONG_VERSION_PY = True +You must add a `versioneer.VCS = "git"` to your `setup.py` before re-running +`setup.py setup_versioneer`. This will enable the use of additional +version-control systems (SVN, etc) in the future. + +## Future Directions + +This tool is designed to make it easily extended to other version-control +systems: all VCS-specific components are in separate directories like +src/git/ . The top-level `versioneer.py` script is assembled from these +components by running make-versioneer.py . In the future, make-versioneer.py +will take a VCS name as an argument, and will construct a version of +`versioneer.py` that is specific to the given VCS. It might also take the +configuration arguments that are currently provided manually during +installation by editing setup.py . Alternatively, it might go the other +direction and include code from all supported VCS systems, reducing the +number of intermediate scripts. + + +## License + +To make Versioneer easier to embed, all its code is dedicated to the public +domain. The `_version.py` that it creates is also in the public domain. +Specifically, both are released under the Creative Commons "Public Domain +Dedication" license (CC0-1.0), as described in +https://creativecommons.org/publicdomain/zero/1.0/ . + +""" + +from __future__ import print_function +try: +    import configparser +except ImportError: +    import ConfigParser as configparser +import errno +import json +import os +import re +import subprocess +import sys + + +class VersioneerConfig: +    """Container for Versioneer configuration parameters.""" + + +def get_root(): +    """Get the project root directory. + +    We require that all commands are run from the project root, i.e. the +    directory that contains setup.py, setup.cfg, and versioneer.py . +    """ +    root = os.path.realpath(os.path.abspath(os.getcwd())) +    setup_py = os.path.join(root, "setup.py") +    versioneer_py = os.path.join(root, "versioneer.py") +    if not (os.path.exists(setup_py) or os.path.exists(versioneer_py)): +        # allow 'python path/to/setup.py COMMAND' +        root = os.path.dirname(os.path.realpath(os.path.abspath(sys.argv[0]))) +        setup_py = os.path.join(root, "setup.py") +        versioneer_py = os.path.join(root, "versioneer.py") +    if not (os.path.exists(setup_py) or os.path.exists(versioneer_py)): +        err = ("Versioneer was unable to run the project root directory. " +               "Versioneer requires setup.py to be executed from " +               "its immediate directory (like 'python setup.py COMMAND'), " +               "or in a way that lets it use sys.argv[0] to find the root " +               "(like 'python path/to/setup.py COMMAND').") +        raise VersioneerBadRootError(err) +    try: +        # Certain runtime workflows (setup.py install/develop in a setuptools +        # tree) execute all dependencies in a single python process, so +        # "versioneer" may be imported multiple times, and python's shared +        # module-import table will cache the first one. So we can't use +        # os.path.dirname(__file__), as that will find whichever +        # versioneer.py was first imported, even in later projects. +        me = os.path.realpath(os.path.abspath(__file__)) +        if os.path.splitext(me)[0] != os.path.splitext(versioneer_py)[0]: +            print("Warning: build in %s is using versioneer.py from %s" +                  % (os.path.dirname(me), versioneer_py)) +    except NameError: +        pass +    return root + + +def get_config_from_root(root): +    """Read the project setup.cfg file to determine Versioneer config.""" +    # This might raise EnvironmentError (if setup.cfg is missing), or +    # configparser.NoSectionError (if it lacks a [versioneer] section), or +    # configparser.NoOptionError (if it lacks "VCS="). See the docstring at +    # the top of versioneer.py for instructions on writing your setup.cfg . +    setup_cfg = os.path.join(root, "setup.cfg") +    parser = configparser.SafeConfigParser() +    with open(setup_cfg, "r") as f: +        parser.readfp(f) +    VCS = parser.get("versioneer", "VCS")  # mandatory + +    def get(parser, name): +        if parser.has_option("versioneer", name): +            return parser.get("versioneer", name) +        return None +    cfg = VersioneerConfig() +    cfg.VCS = VCS +    cfg.style = get(parser, "style") or "" +    cfg.versionfile_source = get(parser, "versionfile_source") +    cfg.versionfile_build = get(parser, "versionfile_build") +    cfg.tag_prefix = get(parser, "tag_prefix") +    if cfg.tag_prefix in ("''", '""'): +        cfg.tag_prefix = "" +    cfg.parentdir_prefix = get(parser, "parentdir_prefix") +    cfg.verbose = get(parser, "verbose") +    return cfg + + +class NotThisMethod(Exception): +    """Exception raised if a method is not valid for the current scenario.""" + +# these dictionaries contain VCS-specific tools +LONG_VERSION_PY = {} +HANDLERS = {} + + +def register_vcs_handler(vcs, method):  # decorator +    """Decorator to mark a method as the handler for a particular VCS.""" +    def decorate(f): +        """Store f in HANDLERS[vcs][method].""" +        if vcs not in HANDLERS: +            HANDLERS[vcs] = {} +        HANDLERS[vcs][method] = f +        return f +    return decorate + + +def run_command(commands, args, cwd=None, verbose=False, hide_stderr=False): +    """Call the given command(s).""" +    assert isinstance(commands, list) +    p = None +    for c in commands: +        try: +            dispcmd = str([c] + args) +            # remember shell=False, so use git.cmd on windows, not just git +            p = subprocess.Popen([c] + args, cwd=cwd, stdout=subprocess.PIPE, +                                 stderr=(subprocess.PIPE if hide_stderr +                                         else None)) +            break +        except EnvironmentError: +            e = sys.exc_info()[1] +            if e.errno == errno.ENOENT: +                continue +            if verbose: +                print("unable to run %s" % dispcmd) +                print(e) +            return None +    else: +        if verbose: +            print("unable to find command, tried %s" % (commands,)) +        return None +    stdout = p.communicate()[0].strip() +    if sys.version_info[0] >= 3: +        stdout = stdout.decode() +    if p.returncode != 0: +        if verbose: +            print("unable to run %s (error)" % dispcmd) +        return None +    return stdout +LONG_VERSION_PY['git'] = '''  # This file helps to compute a version number in source trees obtained from  # git-archive tarball (such as those provided by githubs download-from-tag -# feature). Distribution tarballs (build by setup.py sdist) and build +# feature). Distribution tarballs (built by setup.py sdist) and build  # directories (produced by setup.py build) will contain a much shorter file  # that just contains the computed version number.  # This file is released into the public domain. Generated by -# versioneer-0.7+ (https://github.com/warner/python-versioneer) - -# these strings will be replaced by git during git-archive -git_refnames = "%(DOLLAR)sFormat:%%d%(DOLLAR)s" -git_full = "%(DOLLAR)sFormat:%%H%(DOLLAR)s" +# versioneer-0.16 (https://github.com/warner/python-versioneer) +"""Git implementation of _version.py.""" +import errno +import os +import re  import subprocess  import sys -def run_command(args, cwd=None, verbose=False): -    try: -        # remember shell=False, so use git.exe on windows, not just git -        p = subprocess.Popen(args, stdout=subprocess.PIPE, cwd=cwd) -    except EnvironmentError: -        e = sys.exc_info()[1] + +def get_keywords(): +    """Get the keywords needed to look up the version information.""" +    # these strings will be replaced by git during git-archive. +    # setup.py/versioneer.py will grep for the variable names, so they must +    # each be defined on a line of their own. _version.py will just call +    # get_keywords(). +    git_refnames = "%(DOLLAR)sFormat:%%d%(DOLLAR)s" +    git_full = "%(DOLLAR)sFormat:%%H%(DOLLAR)s" +    keywords = {"refnames": git_refnames, "full": git_full} +    return keywords + + +class VersioneerConfig: +    """Container for Versioneer configuration parameters.""" + + +def get_config(): +    """Create, populate and return the VersioneerConfig() object.""" +    # these strings are filled in when 'setup.py versioneer' creates +    # _version.py +    cfg = VersioneerConfig() +    cfg.VCS = "git" +    cfg.style = "%(STYLE)s" +    cfg.tag_prefix = "%(TAG_PREFIX)s" +    cfg.parentdir_prefix = "%(PARENTDIR_PREFIX)s" +    cfg.versionfile_source = "%(VERSIONFILE_SOURCE)s" +    cfg.verbose = False +    return cfg + + +class NotThisMethod(Exception): +    """Exception raised if a method is not valid for the current scenario.""" + + +LONG_VERSION_PY = {} +HANDLERS = {} + + +def register_vcs_handler(vcs, method):  # decorator +    """Decorator to mark a method as the handler for a particular VCS.""" +    def decorate(f): +        """Store f in HANDLERS[vcs][method].""" +        if vcs not in HANDLERS: +            HANDLERS[vcs] = {} +        HANDLERS[vcs][method] = f +        return f +    return decorate + + +def run_command(commands, args, cwd=None, verbose=False, hide_stderr=False): +    """Call the given command(s).""" +    assert isinstance(commands, list) +    p = None +    for c in commands: +        try: +            dispcmd = str([c] + args) +            # remember shell=False, so use git.cmd on windows, not just git +            p = subprocess.Popen([c] + args, cwd=cwd, stdout=subprocess.PIPE, +                                 stderr=(subprocess.PIPE if hide_stderr +                                         else None)) +            break +        except EnvironmentError: +            e = sys.exc_info()[1] +            if e.errno == errno.ENOENT: +                continue +            if verbose: +                print("unable to run %%s" %% dispcmd) +                print(e) +            return None +    else:          if verbose: -            print("unable to run %%s" %% args[0]) -            print(e) +            print("unable to find command, tried %%s" %% (commands,))          return None      stdout = p.communicate()[0].strip() -    if sys.version >= '3': +    if sys.version_info[0] >= 3:          stdout = stdout.decode()      if p.returncode != 0:          if verbose: -            print("unable to run %%s (error)" %% args[0]) +            print("unable to run %%s (error)" %% dispcmd)          return None      return stdout -import sys -import re -import os.path +def versions_from_parentdir(parentdir_prefix, root, verbose): +    """Try to determine the version from the parent directory name. + +    Source tarballs conventionally unpack into a directory that includes +    both the project name and a version string. +    """ +    dirname = os.path.basename(root) +    if not dirname.startswith(parentdir_prefix): +        if verbose: +            print("guessing rootdir is '%%s', but '%%s' doesn't start with " +                  "prefix '%%s'" %% (root, dirname, parentdir_prefix)) +        raise NotThisMethod("rootdir doesn't start with parentdir_prefix") +    return {"version": dirname[len(parentdir_prefix):], +            "full-revisionid": None, +            "dirty": False, "error": None} -def get_expanded_variables(versionfile_source): + +@register_vcs_handler("git", "get_keywords") +def git_get_keywords(versionfile_abs): +    """Extract version information from the given file."""      # the code embedded in _version.py can just fetch the value of these -    # variables. When used from setup.py, we don't want to import -    # _version.py, so we do it with a regexp instead. This function is not -    # used from _version.py. -    variables = {} +    # keywords. When used from setup.py, we don't want to import _version.py, +    # so we do it with a regexp instead. This function is not used from +    # _version.py. +    keywords = {}      try: -        f = open(versionfile_source,"r") +        f = open(versionfile_abs, "r")          for line in f.readlines():              if line.strip().startswith("git_refnames ="):                  mo = re.search(r'=\s*"(.*)"', line)                  if mo: -                    variables["refnames"] = mo.group(1) +                    keywords["refnames"] = mo.group(1)              if line.strip().startswith("git_full ="):                  mo = re.search(r'=\s*"(.*)"', line)                  if mo: -                    variables["full"] = mo.group(1) +                    keywords["full"] = mo.group(1)          f.close()      except EnvironmentError:          pass -    return variables +    return keywords + -def versions_from_expanded_variables(variables, tag_prefix, verbose=False): -    refnames = variables["refnames"].strip() +@register_vcs_handler("git", "keywords") +def git_versions_from_keywords(keywords, tag_prefix, verbose): +    """Get version information from git keywords.""" +    if not keywords: +        raise NotThisMethod("no keywords at all, weird") +    refnames = keywords["refnames"].strip()      if refnames.startswith("$Format"):          if verbose: -            print("variables are unexpanded, not using") -        return {} # unexpanded, so not in an unpacked git-archive tarball +            print("keywords are unexpanded, not using") +        raise NotThisMethod("unexpanded keywords, not a git-archive tarball")      refs = set([r.strip() for r in refnames.strip("()").split(",")])      # starting in git-1.8.3, tags are listed as "tag: foo-1.0" instead of      # just "foo-1.0". If we see a "tag: " prefix, prefer those. @@ -189,172 +660,350 @@ def versions_from_expanded_variables(variables, tag_prefix, verbose=False):              r = ref[len(tag_prefix):]              if verbose:                  print("picking %%s" %% r) -            return { "version": r, -                     "full": variables["full"].strip() } -    # no suitable tags, so we use the full revision id +            return {"version": r, +                    "full-revisionid": keywords["full"].strip(), +                    "dirty": False, "error": None +                    } +    # no suitable tags, so version is "0+unknown", but full hex is still there      if verbose: -        print("no suitable tags, using full revision id") -    return { "version": variables["full"].strip(), -             "full": variables["full"].strip() } - -def versions_from_vcs(tag_prefix, versionfile_source, verbose=False): -    # this runs 'git' from the root of the source tree. That either means -    # someone ran a setup.py command (and this code is in versioneer.py, so -    # IN_LONG_VERSION_PY=False, thus the containing directory is the root of -    # the source tree), or someone ran a project-specific entry point (and -    # this code is in _version.py, so IN_LONG_VERSION_PY=True, thus the -    # containing directory is somewhere deeper in the source tree). This only -    # gets called if the git-archive 'subst' variables were *not* expanded, -    # and _version.py hasn't already been rewritten with a short version -    # string, meaning we're inside a checked out source tree. +        print("no suitable tags, using unknown + full revision id") +    return {"version": "0+unknown", +            "full-revisionid": keywords["full"].strip(), +            "dirty": False, "error": "no suitable tags"} -    try: -        here = os.path.abspath(__file__) -    except NameError: -        # some py2exe/bbfreeze/non-CPython implementations don't do __file__ -        return {} # not always correct - -    # versionfile_source is the relative path from the top of the source tree -    # (where the .git directory might live) to this file. Invert this to find -    # the root from __file__. -    root = here -    if IN_LONG_VERSION_PY: -        for i in range(len(versionfile_source.split("/"))): -            root = os.path.dirname(root) -    else: -        root = os.path.dirname(here) + +@register_vcs_handler("git", "pieces_from_vcs") +def git_pieces_from_vcs(tag_prefix, root, verbose, run_command=run_command): +    """Get version from 'git describe' in the root of the source tree. + +    This only gets called if the git-archive 'subst' keywords were *not* +    expanded, and _version.py hasn't already been rewritten with a short +    version string, meaning we're inside a checked out source tree. +    """      if not os.path.exists(os.path.join(root, ".git")):          if verbose:              print("no .git in %%s" %% root) -        return {} +        raise NotThisMethod("no .git directory") -    GIT = "git" +    GITS = ["git"]      if sys.platform == "win32": -        GIT = "git.exe" -    stdout = run_command([GIT, "describe", "--tags", "--dirty", "--always"], -                         cwd=root) -    if stdout is None: -        return {} -    if not stdout.startswith(tag_prefix): -        if verbose: -            print("tag '%%s' doesn't start with prefix '%%s'" %% (stdout, tag_prefix)) -        return {} -    tag = stdout[len(tag_prefix):] -    stdout = run_command([GIT, "rev-parse", "HEAD"], cwd=root) -    if stdout is None: -        return {} -    full = stdout.strip() -    if tag.endswith("-dirty"): -        full += "-dirty" -    return {"version": tag, "full": full} - - -def versions_from_parentdir(parentdir_prefix, versionfile_source, verbose=False): -    if IN_LONG_VERSION_PY: -        # We're running from _version.py. If it's from a source tree -        # (execute-in-place), we can work upwards to find the root of the -        # tree, and then check the parent directory for a version string. If -        # it's in an installed application, there's no hope. -        try: -            here = os.path.abspath(__file__) -        except NameError: -            # py2exe/bbfreeze/non-CPython don't have __file__ -            return {} # without __file__, we have no hope -        # versionfile_source is the relative path from the top of the source -        # tree to _version.py. Invert this to find the root from __file__. -        root = here -        for i in range(len(versionfile_source.split("/"))): -            root = os.path.dirname(root) +        GITS = ["git.cmd", "git.exe"] +    # if there is a tag matching tag_prefix, this yields TAG-NUM-gHEX[-dirty] +    # if there isn't one, this yields HEX[-dirty] (no NUM) +    describe_out = run_command(GITS, ["describe", "--tags", "--dirty", +                                      "--always", "--long", +                                      "--match", "%%s*" %% tag_prefix], +                               cwd=root) +    # --long was added in git-1.5.5 +    if describe_out is None: +        raise NotThisMethod("'git describe' failed") +    describe_out = describe_out.strip() +    full_out = run_command(GITS, ["rev-parse", "HEAD"], cwd=root) +    if full_out is None: +        raise NotThisMethod("'git rev-parse' failed") +    full_out = full_out.strip() + +    pieces = {} +    pieces["long"] = full_out +    pieces["short"] = full_out[:7]  # maybe improved later +    pieces["error"] = None + +    # parse describe_out. It will be like TAG-NUM-gHEX[-dirty] or HEX[-dirty] +    # TAG might have hyphens. +    git_describe = describe_out + +    # look for -dirty suffix +    dirty = git_describe.endswith("-dirty") +    pieces["dirty"] = dirty +    if dirty: +        git_describe = git_describe[:git_describe.rindex("-dirty")] + +    # now we have TAG-NUM-gHEX or HEX + +    if "-" in git_describe: +        # TAG-NUM-gHEX +        mo = re.search(r'^(.+)-(\d+)-g([0-9a-f]+)$', git_describe) +        if not mo: +            # unparseable. Maybe git-describe is misbehaving? +            pieces["error"] = ("unable to parse git-describe output: '%%s'" +                               %% describe_out) +            return pieces + +        # tag +        full_tag = mo.group(1) +        if not full_tag.startswith(tag_prefix): +            if verbose: +                fmt = "tag '%%s' doesn't start with prefix '%%s'" +                print(fmt %% (full_tag, tag_prefix)) +            pieces["error"] = ("tag '%%s' doesn't start with prefix '%%s'" +                               %% (full_tag, tag_prefix)) +            return pieces +        pieces["closest-tag"] = full_tag[len(tag_prefix):] + +        # distance: number of commits since tag +        pieces["distance"] = int(mo.group(2)) + +        # commit: short hex revision ID +        pieces["short"] = mo.group(3) + +    else: +        # HEX: no tags +        pieces["closest-tag"] = None +        count_out = run_command(GITS, ["rev-list", "HEAD", "--count"], +                                cwd=root) +        pieces["distance"] = int(count_out)  # total number of commits + +    return pieces + + +def plus_or_dot(pieces): +    """Return a + if we don't already have one, else return a .""" +    if "+" in pieces.get("closest-tag", ""): +        return "." +    return "+" + + +def render_pep440(pieces): +    """Build up version string, with post-release "local version identifier". + +    Our goal: TAG[+DISTANCE.gHEX[.dirty]] . Note that if you +    get a tagged build and then dirty it, you'll get TAG+0.gHEX.dirty + +    Exceptions: +    1: no tags. git_describe was just HEX. 0+untagged.DISTANCE.gHEX[.dirty] +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"] or pieces["dirty"]: +            rendered += plus_or_dot(pieces) +            rendered += "%%d.g%%s" %% (pieces["distance"], pieces["short"]) +            if pieces["dirty"]: +                rendered += ".dirty" +    else: +        # exception #1 +        rendered = "0+untagged.%%d.g%%s" %% (pieces["distance"], +                                          pieces["short"]) +        if pieces["dirty"]: +            rendered += ".dirty" +    return rendered + + +def render_pep440_pre(pieces): +    """TAG[.post.devDISTANCE] -- No -dirty. + +    Exceptions: +    1: no tags. 0.post.devDISTANCE +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"]: +            rendered += ".post.dev%%d" %% pieces["distance"]      else: -        # we're running from versioneer.py, which means we're running from -        # the setup.py in a source tree. sys.argv[0] is setup.py in the root. -        here = os.path.abspath(sys.argv[0]) -        root = os.path.dirname(here) +        # exception #1 +        rendered = "0.post.dev%%d" %% pieces["distance"] +    return rendered + + +def render_pep440_post(pieces): +    """TAG[.postDISTANCE[.dev0]+gHEX] . + +    The ".dev0" means dirty. Note that .dev0 sorts backwards +    (a dirty tree will appear "older" than the corresponding clean one), +    but you shouldn't be releasing software with -dirty anyways. + +    Exceptions: +    1: no tags. 0.postDISTANCE[.dev0] +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"] or pieces["dirty"]: +            rendered += ".post%%d" %% pieces["distance"] +            if pieces["dirty"]: +                rendered += ".dev0" +            rendered += plus_or_dot(pieces) +            rendered += "g%%s" %% pieces["short"] +    else: +        # exception #1 +        rendered = "0.post%%d" %% pieces["distance"] +        if pieces["dirty"]: +            rendered += ".dev0" +        rendered += "+g%%s" %% pieces["short"] +    return rendered + + +def render_pep440_old(pieces): +    """TAG[.postDISTANCE[.dev0]] . + +    The ".dev0" means dirty. + +    Eexceptions: +    1: no tags. 0.postDISTANCE[.dev0] +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"] or pieces["dirty"]: +            rendered += ".post%%d" %% pieces["distance"] +            if pieces["dirty"]: +                rendered += ".dev0" +    else: +        # exception #1 +        rendered = "0.post%%d" %% pieces["distance"] +        if pieces["dirty"]: +            rendered += ".dev0" +    return rendered -    # Source tarballs conventionally unpack into a directory that includes -    # both the project name and a version string. -    dirname = os.path.basename(root) -    if not dirname.startswith(parentdir_prefix): -        if verbose: -            print("guessing rootdir is '%%s', but '%%s' doesn't start with prefix '%%s'" %% -                  (root, dirname, parentdir_prefix)) -        return None -    return {"version": dirname[len(parentdir_prefix):], "full": ""} - -tag_prefix = "%(TAG_PREFIX)s" -parentdir_prefix = "%(PARENTDIR_PREFIX)s" -versionfile_source = "%(VERSIONFILE_SOURCE)s" - -def get_versions(default={"version": "unknown", "full": ""}, verbose=False): -    variables = { "refnames": git_refnames, "full": git_full } -    ver = versions_from_expanded_variables(variables, tag_prefix, verbose) -    if not ver: -        ver = versions_from_vcs(tag_prefix, versionfile_source, verbose) -    if not ver: -        ver = versions_from_parentdir(parentdir_prefix, versionfile_source, -                                      verbose) -    if not ver: -        ver = default -    return ver -''' +def render_git_describe(pieces): +    """TAG[-DISTANCE-gHEX][-dirty]. +    Like 'git describe --tags --dirty --always'. -import subprocess -import sys +    Exceptions: +    1: no tags. HEX[-dirty]  (note: no 'g' prefix) +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"]: +            rendered += "-%%d-g%%s" %% (pieces["distance"], pieces["short"]) +    else: +        # exception #1 +        rendered = pieces["short"] +    if pieces["dirty"]: +        rendered += "-dirty" +    return rendered + + +def render_git_describe_long(pieces): +    """TAG-DISTANCE-gHEX[-dirty]. + +    Like 'git describe --tags --dirty --always -long'. +    The distance/hash is unconditional. + +    Exceptions: +    1: no tags. HEX[-dirty]  (note: no 'g' prefix) +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        rendered += "-%%d-g%%s" %% (pieces["distance"], pieces["short"]) +    else: +        # exception #1 +        rendered = pieces["short"] +    if pieces["dirty"]: +        rendered += "-dirty" +    return rendered + + +def render(pieces, style): +    """Render the given version pieces into the requested style.""" +    if pieces["error"]: +        return {"version": "unknown", +                "full-revisionid": pieces.get("long"), +                "dirty": None, +                "error": pieces["error"]} + +    if not style or style == "default": +        style = "pep440"  # the default + +    if style == "pep440": +        rendered = render_pep440(pieces) +    elif style == "pep440-pre": +        rendered = render_pep440_pre(pieces) +    elif style == "pep440-post": +        rendered = render_pep440_post(pieces) +    elif style == "pep440-old": +        rendered = render_pep440_old(pieces) +    elif style == "git-describe": +        rendered = render_git_describe(pieces) +    elif style == "git-describe-long": +        rendered = render_git_describe_long(pieces) +    else: +        raise ValueError("unknown style '%%s'" %% style) + +    return {"version": rendered, "full-revisionid": pieces["long"], +            "dirty": pieces["dirty"], "error": None} + + +def get_versions(): +    """Get version information or return default if unable to do so.""" +    # I am in _version.py, which lives at ROOT/VERSIONFILE_SOURCE. If we have +    # __file__, we can work backwards from there to the root. Some +    # py2exe/bbfreeze/non-CPython implementations don't do __file__, in which +    # case we can only use expanded keywords. + +    cfg = get_config() +    verbose = cfg.verbose -def run_command(args, cwd=None, verbose=False):      try: -        # remember shell=False, so use git.exe on windows, not just git -        p = subprocess.Popen(args, stdout=subprocess.PIPE, cwd=cwd) -    except EnvironmentError: -        e = sys.exc_info()[1] -        if verbose: -            print("unable to run %s" % args[0]) -            print(e) -        return None -    stdout = p.communicate()[0].strip() -    if sys.version >= '3': -        stdout = stdout.decode() -    if p.returncode != 0: -        if verbose: -            print("unable to run %s (error)" % args[0]) -        return None -    return stdout +        return git_versions_from_keywords(get_keywords(), cfg.tag_prefix, +                                          verbose) +    except NotThisMethod: +        pass +    try: +        root = os.path.realpath(__file__) +        # versionfile_source is the relative path from the top of the source +        # tree (where the .git directory might live) to this file. Invert +        # this to find the root from __file__. +        for i in cfg.versionfile_source.split('/'): +            root = os.path.dirname(root) +    except NameError: +        return {"version": "0+unknown", "full-revisionid": None, +                "dirty": None, +                "error": "unable to find root of source tree"} -import sys -import re -import os.path +    try: +        pieces = git_pieces_from_vcs(cfg.tag_prefix, root, verbose) +        return render(pieces, cfg.style) +    except NotThisMethod: +        pass + +    try: +        if cfg.parentdir_prefix: +            return versions_from_parentdir(cfg.parentdir_prefix, root, verbose) +    except NotThisMethod: +        pass -def get_expanded_variables(versionfile_source): +    return {"version": "0+unknown", "full-revisionid": None, +            "dirty": None, +            "error": "unable to compute version"} +''' + + +@register_vcs_handler("git", "get_keywords") +def git_get_keywords(versionfile_abs): +    """Extract version information from the given file."""      # the code embedded in _version.py can just fetch the value of these -    # variables. When used from setup.py, we don't want to import -    # _version.py, so we do it with a regexp instead. This function is not -    # used from _version.py. -    variables = {} +    # keywords. When used from setup.py, we don't want to import _version.py, +    # so we do it with a regexp instead. This function is not used from +    # _version.py. +    keywords = {}      try: -        f = open(versionfile_source,"r") +        f = open(versionfile_abs, "r")          for line in f.readlines():              if line.strip().startswith("git_refnames ="):                  mo = re.search(r'=\s*"(.*)"', line)                  if mo: -                    variables["refnames"] = mo.group(1) +                    keywords["refnames"] = mo.group(1)              if line.strip().startswith("git_full ="):                  mo = re.search(r'=\s*"(.*)"', line)                  if mo: -                    variables["full"] = mo.group(1) +                    keywords["full"] = mo.group(1)          f.close()      except EnvironmentError:          pass -    return variables +    return keywords + -def versions_from_expanded_variables(variables, tag_prefix, verbose=False): -    refnames = variables["refnames"].strip() +@register_vcs_handler("git", "keywords") +def git_versions_from_keywords(keywords, tag_prefix, verbose): +    """Get version information from git keywords.""" +    if not keywords: +        raise NotThisMethod("no keywords at all, weird") +    refnames = keywords["refnames"].strip()      if refnames.startswith("$Format"):          if verbose: -            print("variables are unexpanded, not using") -        return {} # unexpanded, so not in an unpacked git-archive tarball +            print("keywords are unexpanded, not using") +        raise NotThisMethod("unexpanded keywords, not a git-archive tarball")      refs = set([r.strip() for r in refnames.strip("()").split(",")])      # starting in git-1.8.3, tags are listed as "tag: foo-1.0" instead of      # just "foo-1.0". If we see a "tag: " prefix, prefer those. @@ -379,107 +1028,122 @@ def versions_from_expanded_variables(variables, tag_prefix, verbose=False):              r = ref[len(tag_prefix):]              if verbose:                  print("picking %s" % r) -            return { "version": r, -                     "full": variables["full"].strip() } -    # no suitable tags, so we use the full revision id +            return {"version": r, +                    "full-revisionid": keywords["full"].strip(), +                    "dirty": False, "error": None +                    } +    # no suitable tags, so version is "0+unknown", but full hex is still there      if verbose: -        print("no suitable tags, using full revision id") -    return { "version": variables["full"].strip(), -             "full": variables["full"].strip() } - -def versions_from_vcs(tag_prefix, versionfile_source, verbose=False): -    # this runs 'git' from the root of the source tree. That either means -    # someone ran a setup.py command (and this code is in versioneer.py, so -    # IN_LONG_VERSION_PY=False, thus the containing directory is the root of -    # the source tree), or someone ran a project-specific entry point (and -    # this code is in _version.py, so IN_LONG_VERSION_PY=True, thus the -    # containing directory is somewhere deeper in the source tree). This only -    # gets called if the git-archive 'subst' variables were *not* expanded, -    # and _version.py hasn't already been rewritten with a short version -    # string, meaning we're inside a checked out source tree. +        print("no suitable tags, using unknown + full revision id") +    return {"version": "0+unknown", +            "full-revisionid": keywords["full"].strip(), +            "dirty": False, "error": "no suitable tags"} -    try: -        here = os.path.abspath(__file__) -    except NameError: -        # some py2exe/bbfreeze/non-CPython implementations don't do __file__ -        return {} # not always correct - -    # versionfile_source is the relative path from the top of the source tree -    # (where the .git directory might live) to this file. Invert this to find -    # the root from __file__. -    root = here -    if IN_LONG_VERSION_PY: -        for i in range(len(versionfile_source.split("/"))): -            root = os.path.dirname(root) -    else: -        root = os.path.dirname(here) + +@register_vcs_handler("git", "pieces_from_vcs") +def git_pieces_from_vcs(tag_prefix, root, verbose, run_command=run_command): +    """Get version from 'git describe' in the root of the source tree. + +    This only gets called if the git-archive 'subst' keywords were *not* +    expanded, and _version.py hasn't already been rewritten with a short +    version string, meaning we're inside a checked out source tree. +    """      if not os.path.exists(os.path.join(root, ".git")):          if verbose:              print("no .git in %s" % root) -        return {} +        raise NotThisMethod("no .git directory") -    GIT = "git" +    GITS = ["git"]      if sys.platform == "win32": -        GIT = "git.exe" -    stdout = run_command([GIT, "describe", "--tags", "--dirty", "--always"], -                         cwd=root) -    if stdout is None: -        return {} -    if not stdout.startswith(tag_prefix): -        if verbose: -            print("tag '%s' doesn't start with prefix '%s'" % (stdout, tag_prefix)) -        return {} -    tag = stdout[len(tag_prefix):] -    stdout = run_command([GIT, "rev-parse", "HEAD"], cwd=root) -    if stdout is None: -        return {} -    full = stdout.strip() -    if tag.endswith("-dirty"): -        full += "-dirty" -    return {"version": tag, "full": full} - - -def versions_from_parentdir(parentdir_prefix, versionfile_source, verbose=False): -    if IN_LONG_VERSION_PY: -        # We're running from _version.py. If it's from a source tree -        # (execute-in-place), we can work upwards to find the root of the -        # tree, and then check the parent directory for a version string. If -        # it's in an installed application, there's no hope. -        try: -            here = os.path.abspath(__file__) -        except NameError: -            # py2exe/bbfreeze/non-CPython don't have __file__ -            return {} # without __file__, we have no hope -        # versionfile_source is the relative path from the top of the source -        # tree to _version.py. Invert this to find the root from __file__. -        root = here -        for i in range(len(versionfile_source.split("/"))): -            root = os.path.dirname(root) +        GITS = ["git.cmd", "git.exe"] +    # if there is a tag matching tag_prefix, this yields TAG-NUM-gHEX[-dirty] +    # if there isn't one, this yields HEX[-dirty] (no NUM) +    describe_out = run_command(GITS, ["describe", "--tags", "--dirty", +                                      "--always", "--long", +                                      "--match", "%s*" % tag_prefix], +                               cwd=root) +    # --long was added in git-1.5.5 +    if describe_out is None: +        raise NotThisMethod("'git describe' failed") +    describe_out = describe_out.strip() +    full_out = run_command(GITS, ["rev-parse", "HEAD"], cwd=root) +    if full_out is None: +        raise NotThisMethod("'git rev-parse' failed") +    full_out = full_out.strip() + +    pieces = {} +    pieces["long"] = full_out +    pieces["short"] = full_out[:7]  # maybe improved later +    pieces["error"] = None + +    # parse describe_out. It will be like TAG-NUM-gHEX[-dirty] or HEX[-dirty] +    # TAG might have hyphens. +    git_describe = describe_out + +    # look for -dirty suffix +    dirty = git_describe.endswith("-dirty") +    pieces["dirty"] = dirty +    if dirty: +        git_describe = git_describe[:git_describe.rindex("-dirty")] + +    # now we have TAG-NUM-gHEX or HEX + +    if "-" in git_describe: +        # TAG-NUM-gHEX +        mo = re.search(r'^(.+)-(\d+)-g([0-9a-f]+)$', git_describe) +        if not mo: +            # unparseable. Maybe git-describe is misbehaving? +            pieces["error"] = ("unable to parse git-describe output: '%s'" +                               % describe_out) +            return pieces + +        # tag +        full_tag = mo.group(1) +        if not full_tag.startswith(tag_prefix): +            if verbose: +                fmt = "tag '%s' doesn't start with prefix '%s'" +                print(fmt % (full_tag, tag_prefix)) +            pieces["error"] = ("tag '%s' doesn't start with prefix '%s'" +                               % (full_tag, tag_prefix)) +            return pieces +        pieces["closest-tag"] = full_tag[len(tag_prefix):] + +        # distance: number of commits since tag +        pieces["distance"] = int(mo.group(2)) + +        # commit: short hex revision ID +        pieces["short"] = mo.group(3) +      else: -        # we're running from versioneer.py, which means we're running from -        # the setup.py in a source tree. sys.argv[0] is setup.py in the root. -        here = os.path.abspath(sys.argv[0]) -        root = os.path.dirname(here) +        # HEX: no tags +        pieces["closest-tag"] = None +        count_out = run_command(GITS, ["rev-list", "HEAD", "--count"], +                                cwd=root) +        pieces["distance"] = int(count_out)  # total number of commits -    # Source tarballs conventionally unpack into a directory that includes -    # both the project name and a version string. -    dirname = os.path.basename(root) -    if not dirname.startswith(parentdir_prefix): -        if verbose: -            print("guessing rootdir is '%s', but '%s' doesn't start with prefix '%s'" % -                  (root, dirname, parentdir_prefix)) -        return None -    return {"version": dirname[len(parentdir_prefix):], "full": ""} +    return pieces -import sys -def do_vcs_install(versionfile_source, ipy): -    GIT = "git" +def do_vcs_install(manifest_in, versionfile_source, ipy): +    """Git-specific installation logic for Versioneer. + +    For Git, this means creating/changing .gitattributes to mark _version.py +    for export-time keyword substitution. +    """ +    GITS = ["git"]      if sys.platform == "win32": -        GIT = "git.exe" -    run_command([GIT, "add", "versioneer.py"]) -    run_command([GIT, "add", versionfile_source]) -    run_command([GIT, "add", ipy]) +        GITS = ["git.cmd", "git.exe"] +    files = [manifest_in, versionfile_source] +    if ipy: +        files.append(ipy) +    try: +        me = __file__ +        if me.endswith(".pyc") or me.endswith(".pyo"): +            me = os.path.splitext(me)[0] + ".py" +        versioneer_file = os.path.relpath(me) +    except NameError: +        versioneer_file = "versioneer.py" +    files.append(versioneer_file)      present = False      try:          f = open(".gitattributes", "r") @@ -494,135 +1158,487 @@ def do_vcs_install(versionfile_source, ipy):          f = open(".gitattributes", "a+")          f.write("%s export-subst\n" % versionfile_source)          f.close() -        run_command([GIT, "add", ".gitattributes"]) +        files.append(".gitattributes") +    run_command(GITS, ["add", "--"] + files) + +def versions_from_parentdir(parentdir_prefix, root, verbose): +    """Try to determine the version from the parent directory name. + +    Source tarballs conventionally unpack into a directory that includes +    both the project name and a version string. +    """ +    dirname = os.path.basename(root) +    if not dirname.startswith(parentdir_prefix): +        if verbose: +            print("guessing rootdir is '%s', but '%s' doesn't start with " +                  "prefix '%s'" % (root, dirname, parentdir_prefix)) +        raise NotThisMethod("rootdir doesn't start with parentdir_prefix") +    return {"version": dirname[len(parentdir_prefix):], +            "full-revisionid": None, +            "dirty": False, "error": None}  SHORT_VERSION_PY = """ -# This file was generated by 'versioneer.py' (0.7+) from +# This file was generated by 'versioneer.py' (0.16) from  # revision-control system data, or from the parent directory name of an  # unpacked source archive. Distribution tarballs contain a pre-generated copy  # of this file. -version_version = '%(version)s' -version_full = '%(full)s' -def get_versions(default={}, verbose=False): -    return {'version': version_version, 'full': version_full} +import json +import sys + +version_json = ''' +%s +'''  # END VERSION_JSON + +def get_versions(): +    return json.loads(version_json)  """ -DEFAULT = {"version": "unknown", "full": "unknown"}  def versions_from_file(filename): -    versions = {} +    """Try to determine the version from _version.py if present."""      try: -        f = open(filename) +        with open(filename) as f: +            contents = f.read()      except EnvironmentError: -        return versions -    for line in f.readlines(): -        mo = re.match("version_version = '([^']+)'", line) -        if mo: -            versions["version"] = mo.group(1) -        mo = re.match("version_full = '([^']+)'", line) -        if mo: -            versions["full"] = mo.group(1) -    f.close() -    return versions +        raise NotThisMethod("unable to read _version.py") +    mo = re.search(r"version_json = '''\n(.*)'''  # END VERSION_JSON", +                   contents, re.M | re.S) +    if not mo: +        raise NotThisMethod("no version_json in _version.py") +    return json.loads(mo.group(1)) +  def write_to_version_file(filename, versions): -    f = open(filename, "w") -    f.write(SHORT_VERSION_PY % versions) -    f.close() +    """Write the given version number to the given _version.py file.""" +    os.unlink(filename) +    contents = json.dumps(versions, sort_keys=True, +                          indent=1, separators=(",", ": ")) +    with open(filename, "w") as f: +        f.write(SHORT_VERSION_PY % contents) +      print("set %s to '%s'" % (filename, versions["version"])) -def get_best_versions(versionfile, tag_prefix, parentdir_prefix, -                      default=DEFAULT, verbose=False): -    # returns dict with two keys: 'version' and 'full' -    # -    # extract version from first of _version.py, 'git describe', parentdir. -    # This is meant to work for developers using a source checkout, for users -    # of a tarball created by 'setup.py sdist', and for users of a -    # tarball/zipball created by 'git archive' or github's download-from-tag -    # feature. - -    variables = get_expanded_variables(versionfile_source) -    if variables: -        ver = versions_from_expanded_variables(variables, tag_prefix) -        if ver: -            if verbose: print("got version from expanded variable %s" % ver) -            return ver +def plus_or_dot(pieces): +    """Return a + if we don't already have one, else return a .""" +    if "+" in pieces.get("closest-tag", ""): +        return "." +    return "+" -    ver = versions_from_file(versionfile) -    if ver: -        if verbose: print("got version from file %s %s" % (versionfile, ver)) -        return ver -    ver = versions_from_vcs(tag_prefix, versionfile_source, verbose) -    if ver: -        if verbose: print("got version from git %s" % ver) -        return ver +def render_pep440(pieces): +    """Build up version string, with post-release "local version identifier". -    ver = versions_from_parentdir(parentdir_prefix, versionfile_source, verbose) -    if ver: -        if verbose: print("got version from parentdir %s" % ver) -        return ver +    Our goal: TAG[+DISTANCE.gHEX[.dirty]] . Note that if you +    get a tagged build and then dirty it, you'll get TAG+0.gHEX.dirty + +    Exceptions: +    1: no tags. git_describe was just HEX. 0+untagged.DISTANCE.gHEX[.dirty] +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"] or pieces["dirty"]: +            rendered += plus_or_dot(pieces) +            rendered += "%d.g%s" % (pieces["distance"], pieces["short"]) +            if pieces["dirty"]: +                rendered += ".dirty" +    else: +        # exception #1 +        rendered = "0+untagged.%d.g%s" % (pieces["distance"], +                                          pieces["short"]) +        if pieces["dirty"]: +            rendered += ".dirty" +    return rendered + + +def render_pep440_pre(pieces): +    """TAG[.post.devDISTANCE] -- No -dirty. + +    Exceptions: +    1: no tags. 0.post.devDISTANCE +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"]: +            rendered += ".post.dev%d" % pieces["distance"] +    else: +        # exception #1 +        rendered = "0.post.dev%d" % pieces["distance"] +    return rendered + + +def render_pep440_post(pieces): +    """TAG[.postDISTANCE[.dev0]+gHEX] . + +    The ".dev0" means dirty. Note that .dev0 sorts backwards +    (a dirty tree will appear "older" than the corresponding clean one), +    but you shouldn't be releasing software with -dirty anyways. + +    Exceptions: +    1: no tags. 0.postDISTANCE[.dev0] +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"] or pieces["dirty"]: +            rendered += ".post%d" % pieces["distance"] +            if pieces["dirty"]: +                rendered += ".dev0" +            rendered += plus_or_dot(pieces) +            rendered += "g%s" % pieces["short"] +    else: +        # exception #1 +        rendered = "0.post%d" % pieces["distance"] +        if pieces["dirty"]: +            rendered += ".dev0" +        rendered += "+g%s" % pieces["short"] +    return rendered + + +def render_pep440_old(pieces): +    """TAG[.postDISTANCE[.dev0]] . + +    The ".dev0" means dirty. + +    Eexceptions: +    1: no tags. 0.postDISTANCE[.dev0] +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"] or pieces["dirty"]: +            rendered += ".post%d" % pieces["distance"] +            if pieces["dirty"]: +                rendered += ".dev0" +    else: +        # exception #1 +        rendered = "0.post%d" % pieces["distance"] +        if pieces["dirty"]: +            rendered += ".dev0" +    return rendered + + +def render_git_describe(pieces): +    """TAG[-DISTANCE-gHEX][-dirty]. + +    Like 'git describe --tags --dirty --always'. -    if verbose: print("got version from default %s" % ver) -    return default - -def get_versions(default=DEFAULT, verbose=False): -    assert versionfile_source is not None, "please set versioneer.versionfile_source" -    assert tag_prefix is not None, "please set versioneer.tag_prefix" -    assert parentdir_prefix is not None, "please set versioneer.parentdir_prefix" -    return get_best_versions(versionfile_source, tag_prefix, parentdir_prefix, -                             default=default, verbose=verbose) -def get_version(verbose=False): -    return get_versions(verbose=verbose)["version"] - -class cmd_version(Command): -    description = "report generated version string" -    user_options = [] -    boolean_options = [] -    def initialize_options(self): +    Exceptions: +    1: no tags. HEX[-dirty]  (note: no 'g' prefix) +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        if pieces["distance"]: +            rendered += "-%d-g%s" % (pieces["distance"], pieces["short"]) +    else: +        # exception #1 +        rendered = pieces["short"] +    if pieces["dirty"]: +        rendered += "-dirty" +    return rendered + + +def render_git_describe_long(pieces): +    """TAG-DISTANCE-gHEX[-dirty]. + +    Like 'git describe --tags --dirty --always -long'. +    The distance/hash is unconditional. + +    Exceptions: +    1: no tags. HEX[-dirty]  (note: no 'g' prefix) +    """ +    if pieces["closest-tag"]: +        rendered = pieces["closest-tag"] +        rendered += "-%d-g%s" % (pieces["distance"], pieces["short"]) +    else: +        # exception #1 +        rendered = pieces["short"] +    if pieces["dirty"]: +        rendered += "-dirty" +    return rendered + + +def render(pieces, style): +    """Render the given version pieces into the requested style.""" +    if pieces["error"]: +        return {"version": "unknown", +                "full-revisionid": pieces.get("long"), +                "dirty": None, +                "error": pieces["error"]} + +    if not style or style == "default": +        style = "pep440"  # the default + +    if style == "pep440": +        rendered = render_pep440(pieces) +    elif style == "pep440-pre": +        rendered = render_pep440_pre(pieces) +    elif style == "pep440-post": +        rendered = render_pep440_post(pieces) +    elif style == "pep440-old": +        rendered = render_pep440_old(pieces) +    elif style == "git-describe": +        rendered = render_git_describe(pieces) +    elif style == "git-describe-long": +        rendered = render_git_describe_long(pieces) +    else: +        raise ValueError("unknown style '%s'" % style) + +    return {"version": rendered, "full-revisionid": pieces["long"], +            "dirty": pieces["dirty"], "error": None} + + +class VersioneerBadRootError(Exception): +    """The project root directory is unknown or missing key files.""" + + +def get_versions(verbose=False): +    """Get the project version from whatever source is available. + +    Returns dict with two keys: 'version' and 'full'. +    """ +    if "versioneer" in sys.modules: +        # see the discussion in cmdclass.py:get_cmdclass() +        del sys.modules["versioneer"] + +    root = get_root() +    cfg = get_config_from_root(root) + +    assert cfg.VCS is not None, "please set [versioneer]VCS= in setup.cfg" +    handlers = HANDLERS.get(cfg.VCS) +    assert handlers, "unrecognized VCS '%s'" % cfg.VCS +    verbose = verbose or cfg.verbose +    assert cfg.versionfile_source is not None, \ +        "please set versioneer.versionfile_source" +    assert cfg.tag_prefix is not None, "please set versioneer.tag_prefix" + +    versionfile_abs = os.path.join(root, cfg.versionfile_source) + +    # extract version from first of: _version.py, VCS command (e.g. 'git +    # describe'), parentdir. This is meant to work for developers using a +    # source checkout, for users of a tarball created by 'setup.py sdist', +    # and for users of a tarball/zipball created by 'git archive' or github's +    # download-from-tag feature or the equivalent in other VCSes. + +    get_keywords_f = handlers.get("get_keywords") +    from_keywords_f = handlers.get("keywords") +    if get_keywords_f and from_keywords_f: +        try: +            keywords = get_keywords_f(versionfile_abs) +            ver = from_keywords_f(keywords, cfg.tag_prefix, verbose) +            if verbose: +                print("got version from expanded keyword %s" % ver) +            return ver +        except NotThisMethod: +            pass + +    try: +        ver = versions_from_file(versionfile_abs) +        if verbose: +            print("got version from file %s %s" % (versionfile_abs, ver)) +        return ver +    except NotThisMethod:          pass -    def finalize_options(self): + +    from_vcs_f = handlers.get("pieces_from_vcs") +    if from_vcs_f: +        try: +            pieces = from_vcs_f(cfg.tag_prefix, root, verbose) +            ver = render(pieces, cfg.style) +            if verbose: +                print("got version from VCS %s" % ver) +            return ver +        except NotThisMethod: +            pass + +    try: +        if cfg.parentdir_prefix: +            ver = versions_from_parentdir(cfg.parentdir_prefix, root, verbose) +            if verbose: +                print("got version from parentdir %s" % ver) +            return ver +    except NotThisMethod:          pass -    def run(self): -        ver = get_version(verbose=True) -        print("Version is currently: %s" % ver) - - -class cmd_build(_build): -    def run(self): -        versions = get_versions(verbose=True) -        _build.run(self) -        # now locate _version.py in the new build/ directory and replace it -        # with an updated value -        target_versionfile = os.path.join(self.build_lib, versionfile_build) -        print("UPDATING %s" % target_versionfile) -        os.unlink(target_versionfile) -        f = open(target_versionfile, "w") -        f.write(SHORT_VERSION_PY % versions) -        f.close() -class cmd_sdist(_sdist): -    def run(self): -        versions = get_versions(verbose=True) -        self._versioneer_generated_versions = versions -        # unless we update this, the command will keep using the old version -        self.distribution.metadata.version = versions["version"] -        return _sdist.run(self) - -    def make_release_tree(self, base_dir, files): -        _sdist.make_release_tree(self, base_dir, files) -        # now locate _version.py in the new base_dir directory (remembering -        # that it may be a hardlink) and replace it with an updated value -        target_versionfile = os.path.join(base_dir, versionfile_source) -        print("UPDATING %s" % target_versionfile) -        os.unlink(target_versionfile) -        f = open(target_versionfile, "w") -        f.write(SHORT_VERSION_PY % self._versioneer_generated_versions) -        f.close() +    if verbose: +        print("unable to compute version") + +    return {"version": "0+unknown", "full-revisionid": None, +            "dirty": None, "error": "unable to compute version"} + + +def get_version(): +    """Get the short version string for this project.""" +    return get_versions()["version"] + + +def get_cmdclass(): +    """Get the custom setuptools/distutils subclasses used by Versioneer.""" +    if "versioneer" in sys.modules: +        del sys.modules["versioneer"] +        # this fixes the "python setup.py develop" case (also 'install' and +        # 'easy_install .'), in which subdependencies of the main project are +        # built (using setup.py bdist_egg) in the same python process. Assume +        # a main project A and a dependency B, which use different versions +        # of Versioneer. A's setup.py imports A's Versioneer, leaving it in +        # sys.modules by the time B's setup.py is executed, causing B to run +        # with the wrong versioneer. Setuptools wraps the sub-dep builds in a +        # sandbox that restores sys.modules to it's pre-build state, so the +        # parent is protected against the child's "import versioneer". By +        # removing ourselves from sys.modules here, before the child build +        # happens, we protect the child from the parent's versioneer too. +        # Also see https://github.com/warner/python-versioneer/issues/52 + +    cmds = {} + +    # we add "version" to both distutils and setuptools +    from distutils.core import Command + +    class cmd_version(Command): +        description = "report generated version string" +        user_options = [] +        boolean_options = [] + +        def initialize_options(self): +            pass + +        def finalize_options(self): +            pass + +        def run(self): +            vers = get_versions(verbose=True) +            print("Version: %s" % vers["version"]) +            print(" full-revisionid: %s" % vers.get("full-revisionid")) +            print(" dirty: %s" % vers.get("dirty")) +            if vers["error"]: +                print(" error: %s" % vers["error"]) +    cmds["version"] = cmd_version + +    # we override "build_py" in both distutils and setuptools +    # +    # most invocation pathways end up running build_py: +    #  distutils/build -> build_py +    #  distutils/install -> distutils/build ->.. +    #  setuptools/bdist_wheel -> distutils/install ->.. +    #  setuptools/bdist_egg -> distutils/install_lib -> build_py +    #  setuptools/install -> bdist_egg ->.. +    #  setuptools/develop -> ? + +    # we override different "build_py" commands for both environments +    if "setuptools" in sys.modules: +        from setuptools.command.build_py import build_py as _build_py +    else: +        from distutils.command.build_py import build_py as _build_py + +    class cmd_build_py(_build_py): +        def run(self): +            root = get_root() +            cfg = get_config_from_root(root) +            versions = get_versions() +            _build_py.run(self) +            # now locate _version.py in the new build/ directory and replace +            # it with an updated value +            if cfg.versionfile_build: +                target_versionfile = os.path.join(self.build_lib, +                                                  cfg.versionfile_build) +                print("UPDATING %s" % target_versionfile) +                write_to_version_file(target_versionfile, versions) +    cmds["build_py"] = cmd_build_py + +    if "cx_Freeze" in sys.modules:  # cx_freeze enabled? +        from cx_Freeze.dist import build_exe as _build_exe + +        class cmd_build_exe(_build_exe): +            def run(self): +                root = get_root() +                cfg = get_config_from_root(root) +                versions = get_versions() +                target_versionfile = cfg.versionfile_source +                print("UPDATING %s" % target_versionfile) +                write_to_version_file(target_versionfile, versions) + +                _build_exe.run(self) +                os.unlink(target_versionfile) +                with open(cfg.versionfile_source, "w") as f: +                    LONG = LONG_VERSION_PY[cfg.VCS] +                    f.write(LONG % +                            {"DOLLAR": "$", +                             "STYLE": cfg.style, +                             "TAG_PREFIX": cfg.tag_prefix, +                             "PARENTDIR_PREFIX": cfg.parentdir_prefix, +                             "VERSIONFILE_SOURCE": cfg.versionfile_source, +                             }) +        cmds["build_exe"] = cmd_build_exe +        del cmds["build_py"] + +    # we override different "sdist" commands for both environments +    if "setuptools" in sys.modules: +        from setuptools.command.sdist import sdist as _sdist +    else: +        from distutils.command.sdist import sdist as _sdist + +    class cmd_sdist(_sdist): +        def run(self): +            versions = get_versions() +            self._versioneer_generated_versions = versions +            # unless we update this, the command will keep using the old +            # version +            self.distribution.metadata.version = versions["version"] +            return _sdist.run(self) + +        def make_release_tree(self, base_dir, files): +            root = get_root() +            cfg = get_config_from_root(root) +            _sdist.make_release_tree(self, base_dir, files) +            # now locate _version.py in the new base_dir directory +            # (remembering that it may be a hardlink) and replace it with an +            # updated value +            target_versionfile = os.path.join(base_dir, cfg.versionfile_source) +            print("UPDATING %s" % target_versionfile) +            write_to_version_file(target_versionfile, +                                  self._versioneer_generated_versions) +    cmds["sdist"] = cmd_sdist + +    return cmds + + +CONFIG_ERROR = """ +setup.cfg is missing the necessary Versioneer configuration. You need +a section like: + + [versioneer] + VCS = git + style = pep440 + versionfile_source = src/myproject/_version.py + versionfile_build = myproject/_version.py + tag_prefix = + parentdir_prefix = myproject- + +You will also need to edit your setup.py to use the results: + + import versioneer + setup(version=versioneer.get_version(), +       cmdclass=versioneer.get_cmdclass(), ...) + +Please read the docstring in ./versioneer.py for configuration instructions, +edit setup.cfg, and re-run the installer or 'python versioneer.py setup'. +""" + +SAMPLE_CONFIG = """ +# See the docstring in versioneer.py for instructions. Note that you must +# re-run 'versioneer.py setup' after changing this section, and commit the +# resulting files. + +[versioneer] +#VCS = git +#style = pep440 +#versionfile_source = +#versionfile_build = +#tag_prefix = +#parentdir_prefix = + +"""  INIT_PY_SNIPPET = """  from ._version import get_versions @@ -630,40 +1646,129 @@ __version__ = get_versions()['version']  del get_versions  """ -class cmd_update_files(Command): -    description = "modify __init__.py and create _version.py" -    user_options = [] -    boolean_options = [] -    def initialize_options(self): -        pass -    def finalize_options(self): -        pass -    def run(self): -        ipy = os.path.join(os.path.dirname(versionfile_source), "__init__.py") -        print(" creating %s" % versionfile_source) -        f = open(versionfile_source, "w") -        f.write(LONG_VERSION_PY % {"DOLLAR": "$", -                                   "TAG_PREFIX": tag_prefix, -                                   "PARENTDIR_PREFIX": parentdir_prefix, -                                   "VERSIONFILE_SOURCE": versionfile_source, -                                   }) -        f.close() + +def do_setup(): +    """Main VCS-independent setup function for installing Versioneer.""" +    root = get_root() +    try: +        cfg = get_config_from_root(root) +    except (EnvironmentError, configparser.NoSectionError, +            configparser.NoOptionError) as e: +        if isinstance(e, (EnvironmentError, configparser.NoSectionError)): +            print("Adding sample versioneer config to setup.cfg", +                  file=sys.stderr) +            with open(os.path.join(root, "setup.cfg"), "a") as f: +                f.write(SAMPLE_CONFIG) +        print(CONFIG_ERROR, file=sys.stderr) +        return 1 + +    print(" creating %s" % cfg.versionfile_source) +    with open(cfg.versionfile_source, "w") as f: +        LONG = LONG_VERSION_PY[cfg.VCS] +        f.write(LONG % {"DOLLAR": "$", +                        "STYLE": cfg.style, +                        "TAG_PREFIX": cfg.tag_prefix, +                        "PARENTDIR_PREFIX": cfg.parentdir_prefix, +                        "VERSIONFILE_SOURCE": cfg.versionfile_source, +                        }) + +    ipy = os.path.join(os.path.dirname(cfg.versionfile_source), +                       "__init__.py") +    if os.path.exists(ipy):          try: -            old = open(ipy, "r").read() +            with open(ipy, "r") as f: +                old = f.read()          except EnvironmentError:              old = ""          if INIT_PY_SNIPPET not in old:              print(" appending to %s" % ipy) -            f = open(ipy, "a") -            f.write(INIT_PY_SNIPPET) -            f.close() +            with open(ipy, "a") as f: +                f.write(INIT_PY_SNIPPET)          else:              print(" %s unmodified" % ipy) -        do_vcs_install(versionfile_source, ipy) +    else: +        print(" %s doesn't exist, ok" % ipy) +        ipy = None + +    # Make sure both the top-level "versioneer.py" and versionfile_source +    # (PKG/_version.py, used by runtime code) are in MANIFEST.in, so +    # they'll be copied into source distributions. Pip won't be able to +    # install the package without this. +    manifest_in = os.path.join(root, "MANIFEST.in") +    simple_includes = set() +    try: +        with open(manifest_in, "r") as f: +            for line in f: +                if line.startswith("include "): +                    for include in line.split()[1:]: +                        simple_includes.add(include) +    except EnvironmentError: +        pass +    # That doesn't cover everything MANIFEST.in can do +    # (http://docs.python.org/2/distutils/sourcedist.html#commands), so +    # it might give some false negatives. Appending redundant 'include' +    # lines is safe, though. +    if "versioneer.py" not in simple_includes: +        print(" appending 'versioneer.py' to MANIFEST.in") +        with open(manifest_in, "a") as f: +            f.write("include versioneer.py\n") +    else: +        print(" 'versioneer.py' already in MANIFEST.in") +    if cfg.versionfile_source not in simple_includes: +        print(" appending versionfile_source ('%s') to MANIFEST.in" % +              cfg.versionfile_source) +        with open(manifest_in, "a") as f: +            f.write("include %s\n" % cfg.versionfile_source) +    else: +        print(" versionfile_source already in MANIFEST.in") -def get_cmdclass(): -    return {'version': cmd_version, -            'update_files': cmd_update_files, -            'build': cmd_build, -            'sdist': cmd_sdist, -            } +    # Make VCS-specific changes. For git, this means creating/changing +    # .gitattributes to mark _version.py for export-time keyword +    # substitution. +    do_vcs_install(manifest_in, cfg.versionfile_source, ipy) +    return 0 + + +def scan_setup_py(): +    """Validate the contents of setup.py against Versioneer's expectations.""" +    found = set() +    setters = False +    errors = 0 +    with open("setup.py", "r") as f: +        for line in f.readlines(): +            if "import versioneer" in line: +                found.add("import") +            if "versioneer.get_cmdclass()" in line: +                found.add("cmdclass") +            if "versioneer.get_version()" in line: +                found.add("get_version") +            if "versioneer.VCS" in line: +                setters = True +            if "versioneer.versionfile_source" in line: +                setters = True +    if len(found) != 3: +        print("") +        print("Your setup.py appears to be missing some important items") +        print("(but I might be wrong). Please make sure it has something") +        print("roughly like the following:") +        print("") +        print(" import versioneer") +        print(" setup( version=versioneer.get_version(),") +        print("        cmdclass=versioneer.get_cmdclass(),  ...)") +        print("") +        errors += 1 +    if setters: +        print("You should remove lines like 'versioneer.VCS = ' and") +        print("'versioneer.versionfile_source = ' . This configuration") +        print("now lives in setup.cfg, and should be removed from setup.py") +        print("") +        errors += 1 +    return errors + +if __name__ == "__main__": +    cmd = sys.argv[1] +    if cmd == "setup": +        errors = do_setup() +        errors += scan_setup_py() +        if errors: +            sys.exit(1) | 
