Changeset 638

Timestamp:

10/22/08 16:10:58 (5 years ago)

Author:

rcorsaro

Message:

added warehouse functionality to oforge api and console.

Location:

trunk/plugins/oforgeplugin

Files:

• oforge/api.py (modified) (4 diffs)
• oforge/console.py (modified) (1 diff)
• setup.py (modified) (1 diff)

trunk/plugins/oforgeplugin/oforge/api.py

@@ -70 +70 @@
         props['port'] = \
                 self.config.get('oforge', 'master_warehouse.port', None)
+        props['schema'] = \
+                self.config.get('oforge', 'master_warehouse.schema', None)
         return props
 
@@ -101 +103 @@
         thepath = path.join(base, name)
         env = Environment(thepath)
-        return  env
+        return env
 
     def getconfig(self, name):
@@ -149 +151 @@
         return ("workspace_%s"%(re.sub('(\W+)', '', name))).lower()
 
+    def reload_warehouse_tickets(self, name):
+        env = self.getenv(name)
+        proj_id = self.config.get('oforge', 'proj_id')
+        if proj_id:
+            project = Project.objects.filter(id=proj_id)
+            tickets = Ticket.objects.filter(project=project)
+            tickets.delete()
+            for row in cursor.fetchall():
+                t = Ticket.new(
+                        project=project,
+                        ticket_id=row.id,
+                        type=row.type,
+                        time=int(row.time),
+                        changetime=int(row.changetime),
+                        component=row.component,
+                        severity=row.severity,
+                        priority=row.priority,
+                        owner=row.owner,
+                        report=row.report,
+                        cc=row.cc,
+                        version=row.version,
+                        milestone=row.milestone,
+                        resolution=row.resolution,
+                        summary=row.summary,
+                        description=row.description,
+                        keyword=row.keyword,
+                        status=row.status)
+                t.save()
+
+    def warehouse_project(self, name):
+        env = self.getenv(name)
+        proj_id = env.config.get('oforge', 'proj_id')
+        if proj_id:
+            fields = ('type', 'time', 'changetime', 'component', 'severity', 
+                'priority', 'owner', 'reporter', 'cc', 'version', 'milestone', 
+                'resolution', 'summary', 'description', 'keywords', 'status')
+            names = ','.join(fields)
+            values = ','.join(map(lambda f: "new.%s"%(f), fields))
+            setters = ','.join(map(lambda f: "%s = new.%s"%(f,f), fields))
+            db_cnx = env.get_db_cnx()
+            db_conn_str = env.config.get('trac', 'database')
+            conn_regex = re.compile('^postgres://([^:]+):')
+            db_user = conn_regex.match(db_conn_str).group(1)
+            workspace_prefix = self._canonize_schema_name(name)
+            project = Project.objects.filter(id=proj_id)
+            wh_db_cnx = self.warehouse_db_cnx()
+            try:
+                wh_cursor = wh_db_cnx.cursor()
+                cursor = db_cnx.cursor()
+                schema = env.config.get('oforge', 'postgres_schema')
+                wh_schema = self.config.get('oforge', 'master_warehouse.schema', 'public')
+                wh_cursor.execute("""GRANT USAGE ON 
+                        schema %s TO %s"""%(wh_schema, db_user))
+                wh_cursor.execute("""GRANT INSERT,DELETE,UPDATE ON 
+                        projects_ticket TO %s"""%(db_user))
+                wh_cursor.execute("""GRANT UPDATE,SELECT ON 
+                        projects_ticket_id_seq TO %s"""%(db_user))
+                if not re.match('(.*, ?)?%s($|,)'%(wh_schema), schema):
+                    # Add warehouse schema to the path if it's not added already
+                    wh_cursor.execute("""ALTER USER %s SET search_path TO %s, %s 
+                            """%(db_user, schema, wh_schema))
+                    cursor.execute("SET search_path TO %s, %s"%(schema, wh_schema))
+                wh_db_cnx.commit()
+                cursor.execute("""
+                    CREATE OR REPLACE RULE %s_update_ticket 
+                    AS ON UPDATE TO ticket 
+                    DO ALSO UPDATE projects_ticket 
+                      SET %s 
+                      WHERE ticket_id = old.id AND
+                        project_id = %s"""%(workspace_prefix, setters, 
+                            proj_id))
+                cursor.execute("""
+                    CREATE OR REPLACE RULE %s_insert_ticket 
+                    AS ON INSERT TO ticket 
+                    DO ALSO 
+                      INSERT INTO projects_ticket (project_id, ticket_id, %s)
+                          VALUES(%s, currval('ticket_id_seq'::text), %s)
+                    """%(workspace_prefix, names, proj_id, values))
+                cursor.execute("""
+                    CREATE OR REPLACE RULE %s_delete_ticket 
+                    AS ON DELETE TO ticket 
+                    DO ALSO DELETE FROM projects_ticket 
+                      WHERE ticket_id = old.id AND
+                        project_id = %s"""%(workspace_prefix, proj_id))
+                cursor.execute("SELECT * FROM ticket")
+                db_cnx.commit()
+            except Exception, e:
+                db_cnx.rollback()
+                wh_db_cnx.rollback()
+                raise
+            self.reload_warehouse_tickets(name)
+
     def converttopostgres(self, name):
         """
@@ -167 +261 @@
         config.set('trac', 'database', 
                 self._get_postgres_url(new_user_schema_name, new_password))
+        config.set('oforge', 'postgres_schema', new_user_schema_name)
         config.save()
         new_env = self.getenv(name)

trunk/plugins/oforgeplugin/oforge/console.py

@@ -205 +205 @@
         oforge.converttopostgres(proj)
 
+def warehouse_project():
+    _init_oforge_logger()
+
+    usage = "%prog [options] workspace-shortname  \n - Adds project to data warehouse."
+    parser = OptionParser(usage)
+
+    parser.set_defaults(master_trac_ini=_get_master_trac_ini())
+
+    parser.add_option("--master-trac-ini", dest="master_trac_ini",
+        help="Master trac ini that describes project locations.")
+    (opts, args) = parser.parse_args()
+    proj = args[0]
+    if proj:
+        oforge = OForge(opts.master_trac_ini)
+        oforge.warehouse_project(proj)
+
 def add_to_dashboard():
     _init_oforge_logger()

trunk/plugins/oforgeplugin/setup.py

@@ -40 +40 @@
           'oforge-convert-to-postgres = oforge.console:convert_to_postgres',
           'oforge-add-to-dashboard = oforge.console:add_to_dashboard',
-          'oforge-remove-from-dashboard = oforge.console:remove_from_dashboard'
+          'oforge-remove-from-dashboard = oforge.console:remove_from_dashboard',
+          'oforge-warehouse-project = oforge.console:warehouse_project'
       ]}, 
       zip_safe=True,