Allow setting SVN URL from setup_for_attract CLI command

attract/cli.py

@@ -12,7 +12,8 @@ log = logging.getLogger(__name__)
 
 @manager.command
 @manager.option('-r', '--replace', dest='replace', action='store_true', default=False)
-def setup_for_attract(project_url, replace=False):
+@manager.option('-s', '--svn', dest='svn_url', nargs='?')
+def setup_for_attract(project_url, replace=False, svn_url=None):
     """Adds Attract node types to the project.
 
     Use --replace to replace pre-existing Attract node types
@@ -20,4 +21,4 @@ def setup_for_attract(project_url, replace=False):
     """
 
     authentication.force_cli_user()
-    attract.setup.setup_for_attract(project_url, replace=replace)
+    attract.setup.setup_for_attract(project_url, replace=replace, svn_url=svn_url)

attract/modules.py

@@ -16,26 +16,6 @@ def index():
     return render_template('attract/index.html')
 
 
-@blueprint.route('/subversion/kick')
-def subversion_kick():
-    from . import subversion
-
-    # TODO: each project should have its own SVN server.
-    svn_server_url = 'svn://localhost/agent327'
-    log.info('Re-examining SVN server %s', svn_server_url)
-    client = subversion.obtain(svn_server_url)
-
-    # TODO: last_seen_revision should be stored, probably at the project level.
-    last_seen_revision = 0
-    observer = subversion.CommitLogObserver(client, last_seen_revision=last_seen_revision)
-    observer.fetch_and_observe()
-
-    return jsonify({
-        'previous_last_seen_revision': last_seen_revision,
-        'last_seen_revision': observer.last_seen_revision,
-    })
-
-
 def error_project_not_setup_for_attract():
     return render_template('attract/errors/project_not_setup.html')
 
@@ -90,3 +70,30 @@ def attract_project_view(extra_project_projections=None):
         return wrapper
 
     return decorator
+
+
+@blueprint.route('/<project_url>/subversion/kick')
+@attract_project_view()
+def subversion_kick(project):
+    from . import subversion
+
+    try:
+        pprops = project.extension_props.attract
+    except AttributeError:
+        log.warning("subversion_kick(): Project url=%s doesn't have Attract extension properties.",
+                    project.url)
+        return error_project_not_setup_for_attract()
+
+    svn_server_url = pprops.svn_url  # 'svn://localhost/agent327'
+    log.info('Re-examining SVN server %s', svn_server_url)
+    client = subversion.obtain(svn_server_url)
+
+    # TODO: last_seen_revision should be stored, probably at the project level.
+    last_seen_revision = 0
+    observer = subversion.CommitLogObserver(client, last_seen_revision=last_seen_revision)
+    observer.fetch_and_observe()
+
+    return jsonify({
+        'previous_last_seen_revision': last_seen_revision,
+        'last_seen_revision': observer.last_seen_revision,
+    })

attract/setup.py

@@ -12,6 +12,8 @@ from bson import ObjectId
 from eve.methods.put import put_internal
 from flask import current_app
 
+from . import EXTENSION_NAME
+
 log = logging.getLogger(__name__)
 
 
@@ -53,7 +55,7 @@ def _update_project(project):
         raise RuntimeError("Can't update project %s, issues: %s", project_id, result)
 
 
-def setup_for_attract(project_url, replace=False):
+def setup_for_attract(project_url, replace=False, svn_url=None):
     """Adds Attract node types to the project.
 
     Use --replace to replace pre-existing Attract node types
@@ -105,6 +107,15 @@ def setup_for_attract(project_url, replace=False):
 
         project['node_types'].append(node_type)
 
+    # Set default extension properties. Be careful not to overwrite any properties that
+    # are already there.
+    eprops = project.setdefault('extension_props', {})
+    attract_props = eprops.setdefault(EXTENSION_NAME, {})
+
+    if svn_url:
+        log.info('Setting SVN URL to %s', svn_url)
+        attract_props['svn_url'] = svn_url
+
     _update_project(project)
 
     log.info('Project %s was updated for Attract.', project_url)