Index: infra/services/gnumbd/__main__.py |
diff --git a/infra/services/gnumbd/__main__.py b/infra/services/gnumbd/__main__.py |
index 4f81953c4d46cc23b95a7c4ab80b56ddaa6a789e..b332f965bb8b4a1f83ba8e8b036d26dcf8478e1d 100644 |
--- a/infra/services/gnumbd/__main__.py |
+++ b/infra/services/gnumbd/__main__.py |
@@ -10,12 +10,14 @@ import sys |
import time |
import urlparse |
-LOGGER = logging.getLogger(__name__) |
+from infra.libs import git2 |
+ |
agable
2014/07/25 21:03:39
No newline
iannucci
2014/07/26 00:18:43
Done.
|
+from infra.services.gnumbd import gnumbd |
-from infra.services.gnumbd.inner_loop import ( |
- inner_loop, DEFAULT_CONFIG_REF, DEFAULT_REPO_DIR) |
-from infra.services.gnumbd.support import git, config_ref |
+LOGGER = logging.getLogger(__name__) |
+ |
+DEFAULT_REPO_DIR = 'gnumbd_repos' |
def parse_args(args): # pragma: no cover |
@@ -26,7 +28,7 @@ def parse_args(args): # pragma: no cover |
'Repo URL must use https, git or file protocol.') |
if not parsed.path.strip('/'): |
raise argparse.ArgumentTypeError('URL is missing a path?') |
- return git.Repo(s) |
+ return git2.Repo(s) |
parser = argparse.ArgumentParser('python -m %s' % __package__) |
g = parser.add_mutually_exclusive_group() |
@@ -39,8 +41,6 @@ def parse_args(args): # pragma: no cover |
dest='log_level', help='Make the output really loud.') |
parser.add_argument('--dry_run', action='store_true', |
help='Do not actually push anything.') |
- parser.add_argument('--config_ref', metavar='REF', default=DEFAULT_CONFIG_REF, |
- help='The config ref to use (default: %(default)s)') |
parser.add_argument('--repo_dir', metavar='DIR', default=DEFAULT_REPO_DIR, |
help=('The directory to use for git clones ' |
'(default: %(default)s)')) |
@@ -60,11 +60,12 @@ def parse_args(args): # pragma: no cover |
if e.errno != errno.EEXIST: |
raise |
- return repo, config_ref.ConfigRef(git.Ref(repo, opts.config_ref)) |
+ return repo |
def main(args): # pragma: no cover |
- repo, cref = parse_args(args) |
+ repo = parse_args(args) |
+ cref = gnumbd.GnumbdConfigRef(repo) |
repo.reify() |
loop_count = 0 |
@@ -75,7 +76,7 @@ def main(args): # pragma: no cover |
try: |
- inner_loop(repo, cref) |
+ gnumbd.inner_loop(repo, cref) |
except KeyboardInterrupt: |
raise |
except Exception: |