from autobot.steps import CreateVirtualenv
from buildbot.process.factory import BuildFactory

"""
generic factories 
"""

class VirtualenvFactory(BuildFactory):
  """
  create a virtualenv
  """

  def __init__(self, name='env', hg_sources=()):
    """
    - name : of the virtualenv
    - hg_sources : sources of python packages with setuptools setup.pys
    """
    BuildFactory.__init__(self)
    self.addStep(CreateVirtualenv(name))
    self.addStep(ShellCommand(command=['mkdir', 'src'],
                              directory=name))
    # TODO: set properities:
    # - python location
    # - virtualenv location

    # clone hg repositories
    for hg_source in hg_sources:
      self.addStep(ShellCommand(command=['hg', 'clone', hg_source],
                                directory=name + '/src'))


class PythonSourceFactory(VirtualenvFactory):
  """
  setup several python packages
  """

  def __init__(self, name='env', hg_sources=()):

    # setup the environment
    VirtualenvFactory.__init__(self, name=name, hg_sources=hg_sources)

    # install the packages


# python sources:
# [{'source': 'http://hg.mozilla.org/....',
#   'type': 'hg', # or git,
#  }]