From a1053093bb038ca96d030f9d983fa75e7816c822 Mon Sep 17 00:00:00 2001

From: Stephen Gallagher <sgallagh@redhat.com>

Date: Tue, 4 Jun 2013 14:47:38 -0400

Subject: [PATCH 1/2] Store installation paths to sitelist file during 'rb-site

 install'

---

 reviewboard/cmdline/rbsite.py | 76 +++++++++++++++++++++++++++++++++++++++++--

 1 file changed, 74 insertions(+), 2 deletions(-)

diff --git a/reviewboard/cmdline/rbsite.py b/reviewboard/cmdline/rbsite.py

index 6b72c83bf7a4aec16d9bada270571ab4af822467..ed7f1f575490c17219b69ad44af137c3dface38b 100755

--- a/reviewboard/cmdline/rbsite.py

+++ b/reviewboard/cmdline/rbsite.py

@@ -18,6 +18,8 @@ from reviewboard import get_version_string

 DOCS_BASE = "http://www.reviewboard.org/docs/manual/dev/"

+SITELIST_FILE_UNIX = "/etc/reviewboard/sites"

+

 # See if GTK is a possibility.

 try:

@@ -639,6 +641,58 @@ class Site(object):

         fp.close()

+class SiteList(object):

+    """Maintains the list of sites installed on the system."""

+    def __init__(self, path):

+        self.path = path

+

+        # Read the list in as a unique set.

+        # This way, we can easily eliminate duplicates.

+        self.sites = set()

+

+        if os.path.exists(self.path):

+            f = open(self.path, 'r')

+

+            for line in f:

+                site = line.strip()

+

+                # Verify that this path exists on the system

+                # And add it to the dictionary.

+                if os.path.exists(site):

+                    self.sites.add(site)

+

+            f.close()

+

+    def add_site(self, site_path):

+        self.sites.add(site_path)

+

+        # Write all of the sites back to the file.

+        # Sort keys to ensure consistent order.

+        ordered_sites = list(self.sites)

+        ordered_sites.sort()

+

+        # Create the parent directory of the site

+        # if it doesn't already exist

+        if not os.path.exists(os.path.dirname(self.path)):

+            # Create the parent directory with read-write

+            # permissions for user but read and execute

+            # only for others.

+            try:

+                os.makedirs(os.path.dirname(self.path), 0755)

+            except error:

+                # We shouldn't consider this an abort-worthy error

+                # We'll warn the user and just complete setup

+                print "WARNING: Could not save site to sitelist %s" % self.path

+                return

+

+        f = open(self.path, "w")

+

+        for site in ordered_sites:

+            f.write("%s\n" % site)

+

+        f.close()

+

+

 class UIToolkit(object):

     """

     An abstract class that forms the basis for all UI interaction.

@@ -1332,12 +1386,13 @@ class InstallCommand(Command):

     needs_ui = True

     def add_options(self, parser):

-        isWin = (platform.system() == "Windows")

+        is_windows = platform.system() == "Windows"

         group = OptionGroup(parser, "'install' command",

                             self.__doc__.strip())

         group.add_option("--copy-media", action="store_true",

-                         dest="copy_media", default=isWin,

+                         dest="copy_media",

+                         default=is_windows,

                          help="copy media files instead of symlinking")

         group.add_option("--noinput", action="store_true", default=False,

@@ -1384,6 +1439,13 @@ class InstallCommand(Command):

         group.add_option("--admin-email",

                          help="the site administrator's e-mail address")

+        # UNIX-specific arguments

+        if not is_windows:

+            group.add_option("--sitelist",

+                             default=SITELIST_FILE_UNIX,

+                             help="the path to a file storing a list of "

+                                  "installed sites")

+

         parser.add_option_group(group)

     def run(self):

@@ -1413,6 +1475,7 @@ class InstallCommand(Command):

             self.ask_web_server_type()

             self.ask_python_loader()

             self.ask_admin_user()

+            # Do not ask for sitelist file, it should not be common.

         self.show_install_status()

         self.show_finished()

@@ -1759,6 +1822,15 @@ class InstallCommand(Command):

         siteconfig.set("site_admin_email", site.admin_email)

         siteconfig.save()

+        if platform.system() != 'Windows':

+            abs_sitelist = os.path.abspath(site.sitelist)

+

+            # Add the site to the sitelist file.

+            print("Saving site %s to the sitelist %s\n" % (

+                  site.install_dir, abs_sitelist))

+            sitelist = SiteList(abs_sitelist)

+            sitelist.add_site(site.install_dir)

+

 class UpgradeCommand(Command):

     """

-- 

1.8.3.1