PostgresqlPatch: pgsql_patch_05.diff

scripts/trac-admin

@@ -27 +27 @@
-import time
-import cmd
-import shlex
-import sqlite
-import StringIO
+import traceback
-
-from trac import util
-from trac import sync
@@ -92 +92 @@
-            return 0
-        return 1
-        
-
+, db_str
-        try:
-create=1
+1, db_str
-            return self.__env
-        except Exception, e:
-            print 'Failed to create environment.', e
@@ -111 +111 @@
-
-    def db_execsql (self, sql, cursor=None):
-        data = []
+        row = None
-        if not cursor:
-            cnx=self.db_open()
-            cursor = cnx.cursor()
@@ -118 +119 @@
-            cnx = None
-        cursor.execute(sql)
-        while 1:
-
+
+
+
+
-            if row == None:
-                break
-            data.append(row)
@@ -457 +461 @@
-
-    ## Initenv
-    _help_initenv = [('initenv', 'Create and initialize a new environment interactively'),
-
+ <dbms> [dbms options]
-                      'Create and initialize a new environment from arguments')]
-
-    def do_initdb(self, line):
@@ -490 +494 @@
-        dt = trac.siteconfig.__default_templates_dir__
-        prompt = 'Templates directory [%s]> ' % dt
-        returnvals.append(raw_input(prompt) or dt)
+        print
+        print " Please enter the database in which you wish to store Trac data."
+        print " Default is 'sqlite', but others are supported:"
+        print " 'sqlite' - SQLite       http://www.sqlite.org"
+        print " 'pgsql'  - PostgreSQL   http://www.postgresql.org"
+        print
+        ddb = "sqlite"
+        prompt =  'database system [%s]> ' % ddb
+        dbms = raw_input(prompt) or ddb
+        returnvals.append(dbms)
+        
+        # PostgreSQL - additional questions
+        if dbms == "pgsql":
+            print
+            print " Please enter the host of your PostgreSQL database."
+            print " Default is 'localhost'"
+            print
+            host = "localhost"
+            prompt = "PostgreSQL host [%s]> " % host
+            returnvals.append(raw_input(prompt) or host)
+            
+            print
+            print " Please enter the port of your PostgreSQL database."
+            print " Default is '5432'"
+            print
+            port = "5432"
+            prompt = "PostgreSQL port [%s]> " % port
+            returnvals.append(raw_input(prompt) or port)
+            
+            print
+            print " Please enter the name of your PostgreSQL database."
+            print " Default is 'trac'"
+            print " Note: This database should not exist, as trac-admin will" \
+                  " attempt to create it."
+            print
+            name = "trac"
+            prompt = "PostgreSQL database [%s]> " % name
+            returnvals.append(raw_input(prompt) or name)
+            
+            print
+            print " Please enter the user of your PostgreSQL database."
+            print " Default is 'trac'"
+            print
+            user = "trac"
+            prompt = "PostgreSQL user [%s]> " % user
+            returnvals.append(raw_input(prompt) or user)
+            
+            print
+            print " Please enter the password of your PostgreSQL database."
+            print " Default is 'trac'"
+            print
+            password = "trac"
+            prompt = "PostgreSQL password [%s]> " % password
+            returnvals.append(raw_input(prompt) or password)
+            
-        return returnvals
-         
-    def do_initenv(self, line):
@@ -500 +559 @@
-        project_name = None
-        repository_dir = None
-        templates_dir = None
+        db_str = None
-        if len(arg) == 1:
-            returnvals = self.get_initenv_args()
-            project_name = returnvals[0]
-            repository_dir = returnvals[1]
-            templates_dir = returnvals[2]
-
+
+
-            print 'Wrong number of arguments to initenv %d' % len(arg)
-            return
-        else:
-            project_name = arg[0]
-            repository_dir = arg[1]
-            templates_dir = arg[2]
+            db_str = arg[3]
+        
+        # sqlite-specific stuff
+        if db_str.lower() == "sqlite":
+            db_str = 'sqlite:"db/trac.db",timeout=10000'
+        
+        # postgres-specific stuff
+        if db_str.lower() == "pgsql":
+            if len(arg) == 1:
+                host = returnvals[4]
+                port = returnvals[5]
+                database = returnvals[6]
+                user = returnvals[7]
+                password = returnvals[8]
+            elif len(arg) != 9:
+                print 'Wrong number of arguments to initenv %d' % len(arg)
+                print 'For PostgreSQL (after pgsql): <host> <port> <database>' \
+                      ' <user> <password>'
+                return
+            else:
+                host = arg[4]
+                port = arg[5]
+                database = arg[6]
+                user = arg[7]
+                password = arg[8]
+            db_str = "pgsql:\"\",host='%s:%s',database='%s',user='%s'," \
+                     "password='%s'" \
+                     % (host, port, database, user, password)
+            createdb_str = "createdb --host=%s --port=%s --owner=%s " \
+                           "--username=%s %s\n" \
+                           % (host, port, user, user, database)
+            print createdb_str
+            try:
+                # TODO: this is not cross-platform.
+                # for Windows, look at win32pipe.popen()
+                os.popen(createdb_str)
+            except Exception, e:
+                print "Error creating PostgreSQL database: %s" % e
+                print "command: %s" % createdb_str
+        
-        from svn import util, repos, core
-        core.apr_initialize()
-        pool = core.svn_pool_create(None)
@@ -530 +631 @@
-            return
-        try:
-            print 'Creating and Initializing Project'
-
+db_str
-            cnx = self.__env.get_db_cnx()
-            print ' Inserting default data'
-            self.__env.insert_default_data()
@@ -669 +770 @@
-                self.do_help ('wiki')
-        except Exception, e:
-            print 'Wiki %s failed:' % arg[0], e
+            print traceback.print_exc()
+            
-
-    def _do_wiki_list(self):
-        data = self.db_execsql('SELECT name,max(version),time'
@@ -694 +797 @@
-        data = data.replace("'", "''") # Escape ' for safe SQL
-        f.close()
-        
-
-
-
+
+
+
+
-               % {'title':title,
-                  'time':int(time.time()),
-                  'author':'trac',
@@ -1007 +1111 @@
-                    print "Upgrade: Backup of old database saved in " \
-                          "%s/db/trac.db.%i.bak" % (self.envname, curr)
-                else:
-a
+e
-                self.__env.upgrade(do_backup)
-            else:
-                print "Upgrade: Database is up to date, no upgrade necessary."

trac/db_default.py

@@ -30 +30 @@
-    result = []
-    i = 1
-    for r in reps:
-i, 
+
-        i = i + 1
-    return result
-
@@ -174 +174 @@
-CREATE INDEX session_idx        ON session(sid,var_name);
-"""
-
+schema_pgsql = """
+CREATE TABLE revision (
+        rev             integer,
+        time            integer,
+        author          text,
+        message         text,
+        CONSTRAINT revision_pkey PRIMARY KEY (rev)
+);
+
+CREATE TABLE node_change (
+        rev             integer,
+        name            text,
+        change          char(1),
+        CONSTRAINT node_change_pkey PRIMARY KEY (rev, name, change)
+);
+
+CREATE TABLE auth_cookie (
+        cookie          text,
+        name            text,
+        ipnr            text,
+        time            integer,
+        CONSTRAINT auth_cookie_pkey PRIMARY KEY(cookie, name, ipnr)
+);
+
+CREATE TABLE enum (
+        type            text,
+        name            text,
+        value           integer,
+        CONSTRAINT enum_pkey PRIMARY KEY(name,type)
+);
+
+CREATE TABLE system (
+        name            text,
+        value           text,
+        CONSTRAINT system_pkey PRIMARY KEY(name)
+);
+
+CREATE TABLE lock (
+        name            text,
+        owner           text,
+        ipnr            text,
+        time            integer,
+        CONSTRAINT lock_pkey PRIMARY KEY(name)
+);
+
+--CREATE SEQUENCE ticket_id_seq;
+CREATE TABLE ticket (
+        id              serial,
+        time            integer,        -- the time it was created
+        changetime      integer,
+        component       text,
+        severity        text,
+        priority        text,
+        owner           text,           -- who is this ticket assigned to
+        reporter        text,
+        cc              text,           -- email addresses to notify
+        url             text,           -- url related to this ticket
+        version         text,           -- 
+        milestone       text,           -- 
+        status          text,
+        resolution      text,
+        summary         text,           -- one-line summary
+        description     text,           -- problem description (long)
+        keywords        text,
+        CONSTRAINT ticket_pkey PRIMARY KEY(id)
+);
+
+CREATE TABLE ticket_change (
+        ticket          integer,
+        time            integer,
+        author          text,
+        field           text,
+        oldvalue        text,
+        newvalue        text,
+        CONSTRAINT ticket_change_pkey PRIMARY KEY(ticket, time, field)
+);
+
+CREATE TABLE ticket_custom (
+       ticket           integer,
+       name             text,
+       value            text,
+       CONSTRAINT ticket_custom_pkey PRIMARY KEY(ticket,name)
+);
+
+--CREATE SEQUENCE report_id_seq;
+CREATE TABLE report (
+        id              serial,
+        author          text,
+        title           text,
+        sql             text,
+        description     text,
+        CONSTRAINT report_pkey PRIMARY KEY(id)
+);
+
+CREATE TABLE permission (
+        username        text,           -- 
+        action          text,           -- allowable activity
+        CONSTRAINT permission_pkey PRIMARY KEY(username,action)
+);
+
+CREATE TABLE component (
+         name            text,
+         owner           text,
+         CONSTRAINT component_pkey PRIMARY KEY(name)
+);
+
+CREATE TABLE milestone (
+         name            text,
+         due             integer, -- Due date/time
+         completed       integer, -- Completed date/time
+         description     text,
+         CONSTRAINT milestone_pkey PRIMARY KEY(name)
+);
+
+CREATE TABLE version (
+         name            text,
+         time            integer,
+         CONSTRAINT version_pkey PRIMARY KEY(name)
+);
+
+CREATE TABLE wiki (
+         name            text,
+         version         integer,
+         time            integer,
+         author          text,
+         ipnr            text,
+         text            text,
+         comment         text,
+         readonly        integer,
+         CONSTRAINT wiki_pkey PRIMARY KEY(name,version)
+);
+
+CREATE TABLE attachment (
+         type            text,
+         id              text,
+         filename        text,
+         size            integer,
+         time            integer,
+         description     text,
+         author          text,
+         ipnr            text,
+         CONSTRAINT attachment_pkey PRIMARY KEY(type,id,filename)
+);
+
+CREATE TABLE session (
+         sid             text,
+         username        text,
+         var_name        text,
+         var_value       text,
+         CONSTRAINT session_pkey PRIMARY KEY(sid,var_name)
+);
+
+"""
+
-##
-## Default Reports
-##
@@ -295 +449 @@
-  FROM ticket t,enum p
-  WHERE p.name=t.priority AND p.type='priority'
-  ORDER BY (milestone IS NULL), milestone DESC, (status = 'closed'), 
-modified ELSE -
+changetime ELSE (-1)*
-"""),
-#----------------------------------------------------------------------------
-('My Tickets',
@@ -406 +560 @@
-             ('name', 'value'),
-               (('database_version', str(db_version)),)),
-           ('report',
-id', '
+
-               __mkreports(reports)))
-
-default_config = \
- (('trac', 'htdocs_location', '/trac/'),
-  ('trac', 'repository_dir', '/var/svn/myrep'),
-  ('trac', 'templates_dir', '/usr/lib/trac/templates'),
-db/trac.db
+"db/trac.db",timeout=10000
-  ('trac', 'default_charset', 'iso-8859-15'),
-  ('logging', 'log_type', 'none'),
-  ('logging', 'log_file', 'trac.log'),

trac/Milestone.py

@@ -157 +157 @@
-        cursor = self.db.cursor()
-        self.log.debug("Creating new milestone '%s'" % name)
-        cursor.execute("INSERT INTO milestone (name,due,completed,description) "
-d,%d
+s,%s
-                       description)
-        self.db.commit()
-        self.req.redirect(self.env.href.milestone(name))
@@ -196 +196 @@
-                          'associated with milestone %s' % id)
-            cursor.execute("UPDATE ticket SET milestone = %s "
-                           "WHERE milestone = %s", name, id)
-d
-d
+s
+s
-                           name, due, completed, description, id)
-            self.db.commit()
-            self.req.redirect(self.env.href.milestone(name))
@@ -209 +209 @@
-        groups = []
-        if by in ['status', 'resolution', 'severity', 'priority']:
-            cursor.execute("SELECT name FROM enum WHERE type = %s "
-IFNULL
+COALESCE
-        elif by in ['component', 'milestone', 'version']:
-            cursor.execute("SELECT name FROM %s "
-IFNULL
+COALESCE
-        elif by == 'owner':
-            cursor.execute("SELECT DISTINCT owner AS name FROM ticket "
-                           "ORDER BY owner")

trac/Search.py

@@ -117 +117 @@
-
-        cursor = self.db.cursor ()
-
+        # ugly 'cast' hack for sqlite vs other dbms
+        # and UNION ALL requiring like column datatypes
+        data_changeset = 'rev'
+        data_tickets = 'a.id'
+        dbms = self.env.dbms
+        if dbms != 'sqlite':
+            data_changeset = 'cast(rev as text)'
+            data_tickets = 'cast(a.id as text)'
+
-        q = []
-        if changeset:
-            q.append('SELECT 1 as type, message AS title, message, author, '
-rev
+%s
-                     ' FROM revision WHERE %s OR %s' % 
-
+
+
-                      self.query_to_sql(query, 'author')))
-        if tickets:
-            q.append('SELECT DISTINCT 2 as type, a.summary AS title, '
-                     ' a.description AS message, a.reporter AS author, '
-
-
-
-
-
+
+
+
+
+
+
-                       self.query_to_sql(query, 'summary'),
-                       self.query_to_sql(query, 'keywords'),
-                       self.query_to_sql(query, 'description'),

trac/Query.py

@@ -154 +154 @@
-        sql.append("\nFROM ticket")
-        for k in [k for k in cols if k in custom_fields]:
-           sql.append("\n  LEFT OUTER JOIN ticket_custom AS %s ON " \
-
+ tc
-
-        for col in [c for c in ['status', 'resolution', 'priority', 'severity']
-                    if c == self.order or c == self.group]:
-            sql.append("\n  LEFT OUTER JOIN (SELECT name AS %s_name, " \
-                                            "value AS %s_value " \
-
+ e
-                       " ON %s_name=%s" % (col, col, col, col, col))
-        for col in [c for c in ['milestone', 'version']
-                    if c == self.order or c == self.group]:
-            time_col = col == 'milestone' and 'due' or 'time'
-            sql.append("\n  LEFT OUTER JOIN (SELECT name AS %s_name, " \
-
+ a
-                       " ON %s_name=%s" % (col, time_col, col, col, col, col))
-
-        def get_constraint_sql(name, value, mode, neg):
-            value = sql_escape(value[len(mode and '!' or '' + mode):])
-            if mode == '~' and value:
-IFNULL
+COALESCE
-                       name, neg and 'NOT ' or '', value)
-            elif mode == '^' and value:
-IFNULL
+COALESCE
-                       name, neg and 'NOT ' or '', value)
-            elif mode == '$' and value:
-IFNULL
+COALESCE
-                       name, neg and 'NOT ' or '', value)
-            elif mode == '':
-IFNULL
+COALESCE
-
-        clauses = []
-        for k, v in self.constraints.items():
@@ -195 +195 @@
-            # Special case for exact matches on multiple values
-            if not mode and len(v) > 1:
-                inlist = ",".join(["'" + sql_escape(val[neg and 1 or 0:]) + "'" for val in v])
-IFNULL
+COALESCE
-            elif len(v) > 1:
-                constraint_sql = [get_constraint_sql(k, val, mode, neg) for val in v]
-                if neg:
@@ -214 +214 @@
-        if self.group and self.group != self.order:
-            order_cols.insert(0, (self.group, self.groupdesc))
-        for col, desc in order_cols:
-
-
+
+
+
+
+
+
+
+
-            else:
-
+
+
+
+
-            if col in ['status', 'resolution', 'priority', 'severity']:
-                if desc:
-                    sql.append("%s_value DESC" % col)
@@ -225 +234 @@
-                    sql.append("%s_value" % col)
-            elif col in ['milestone', 'version']:
-                if desc:
-IFNULL
+COALESCE
-                               % (col, col, col))
-                else:
-IFNULL
+COALESCE
-                               % (col, col, col))
-            else:
-                if desc:

trac/Timeline.py

@@ -142 +142 @@
-                if max_node != 0:
-                    cursor_node = self.db.cursor ()
-                    cursor_node.execute("SELECT name, change "
-d" %
+s",
-                    node_list = ''
-                    node_data = ''
-                    node_count = 0;

trac/Report.py

@@ -110 +110 @@
-    def create_report(self, title, description, sql):
-        self.perm.assert_permission(perm.REPORT_CREATE)
-
+        dbms = self.env.get_dbms()
-        cursor = self.db.cursor()
-
-
-
+
+
+
+
+
+
+
+
+
+
-        self.db.commit()
-        self.req.redirect(self.env.href.report(id))
-
@@ -140 +148 @@
-
-        # FIXME: fetchall should probably not be used.
-        info = cursor.fetchall()
-rs.col_defs
+description
-        # Escape the values so that they are safe to have as html parameters
-        #info = map(lambda row: map(lambda x: escape(x), row), info)
-

trac/sync.py

@@ -35 +35 @@
-              (util.SVN_VER_MAJOR, util.SVN_VER_MINOR, util.SVN_VER_MICRO)
-
-    cursor = db.cursor()
-ifnull
+COALESCE
-    youngest_stored =  int(cursor.fetchone()[0])
-    max_rev = fs.youngest_rev(fs_ptr, pool)
-    num = max_rev - youngest_stored

trac/Roadmap.py

@@ -44 +44 @@
-        if show == 'all':
-            icalhref += '&show=all'
-            query = "SELECT name,due,completed,description FROM milestone " \
-IFNULL
-IFNULL
+COALESCE
+COALESCE
-        else:
-            self.req.hdf.setValue('roadmap.showall', '1')
-            query = "SELECT name,due,completed,description FROM milestone " \
-IFNULL
-IFNULL
-IFNULL
+COALESCE
+COALESCE
+COALESCE
-
-        if self.req.authname and self.req.authname != 'anonymous':
-            icalhref += '&user=' + self.req.authname
@@ -176 +176 @@
-                if ticket['status'] == 'closed':
-                    cursor = self.db.cursor()
-                    cursor.execute("SELECT time FROM ticket_change "
-i
+s
-                                   "ORDER BY time desc LIMIT 1", ticket['id'])
-                    row = cursor.fetchone()
-                    if row: write_utctime('COMPLETED', localtime(row['time']))

trac/Session.py

@@ -103 +103 @@
-    def get_session(self, sid):
-        self.sid = sid
-        curs = self.db.cursor()
-
-self.sid
+ 
+(self.sid,)
-        rows = curs.fetchall()
-        if (not rows                              # No session data yet
-            or rows[0][0] == 'anonymous'          # Anon session
@@ -138 +138 @@
-                self.purge_expired() 
-            curs.execute('INSERT INTO session(sid,username,var_name,var_value)'
-                         ' VALUES(%s,%s,%s,%s)',
-self.sid, self.req.authname, key, val
+(self.sid, self.req.authname, key, val,)
-        else:
-            curs.execute('UPDATE session SET username=%s,var_value=%s'
-                         ' WHERE sid=%s AND var_name=%s',
-self.req.authname, val, self.sid, key
+(self.req.authname, val, self.sid, key,)
-        self.db.commit()
-        self.vars[key] = val
-
@@ -172 +172 @@
-        curs = self.db.cursor()
-        curs.execute("DELETE FROM session WHERE sid IN"
-                     " (SELECT sid FROM session WHERE var_name='mod_time'"
-i
+s
-        self.db.commit()
-

trac/auth.py

@@ -32 +32 @@
-            cookie = req.incookie['trac_auth'].value
-            cursor.execute ("SELECT name FROM auth_cookie "
-                            "WHERE cookie=%s AND ipnr=%s"
-cookie, req.remote_addr
-if cursor.rowcount >= 1
+ (cookie, req.remote_addr,)
+try
-                self.authname = cursor.fetchone()[0]
+            except:
+                pass
-
-    def login(self, req):
-        cursor = self.db.cursor ()
-        cookie = util.hex_entropy()
-        cursor.execute ("INSERT INTO auth_cookie (cookie, name, ipnr, time)" +
-d
-
-
+s
+(
+,)
-        self.db.commit ()
-        self.authname = req.remote_user
-        req.outcookie['trac_auth'] = cookie
@@ -51 +53 @@
-    def logout(self):
-        cursor = self.db.cursor ()
-        cursor.execute ("DELETE FROM auth_cookie WHERE name=%s",
-self.authname
+(self.authname,)
-        self.db.commit ()

trac/Wiki.py

@@ -59 +59 @@
-        if version:
-            cursor.execute ('SELECT version, text, readonly FROM wiki '
-                            'WHERE name=%s AND version=%s',
-name, version
+(name, version,)
-        else:
-            cursor.execute ('SELECT version, text, readonly FROM wiki '
-
+
+
-        row = cursor.fetchone()
-        if row:
-            self.new = 0

trac/Environment.py

@@ -34 +34 @@
-import Logging
-import Mimeview
-import unicodedata
+import types
-
-import sqlite
-
-db_version = db_default.db_version
-
-class Environment:
@@ -46 +45 @@
-    A Trac environment consists of a directory structure containing
-    among other things:
-     * a configuration file.
-sqlite 
+
-     * Project specific templates and wiki macros.
-     * wiki and ticket attachments.
-    """
-
+, db_str=None
-        self.path = path
-        if create:
-
+db_str
-        self.verify()
-        self.load_config()
-        try: # Use binary I/O on Windows
@@ -64 +63 @@
-            pass
-        self.setup_log()
-        self.setup_mimeviewer()
+        self.dbms = self.get_dbms()
-
-    def verify(self):
-        """Verifies that self.path is a compatible trac environment"""
@@ -71 +71 @@
-        assert fd.read(26) == 'Trac Environment Version 1'
-        fd.close()
-
-
-
-
-
-
-
+
+
+
+
+
+
+
+
+
-        
-
-
-
-
-
-
-
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-        def _create_file(fname, data=None):
-            fd = open(fname, 'w')
-            if data: fd.write(data)
@@ -128 +196 @@
-""")
-        # Create default database
-        os.mkdir(os.path.join(self.path, 'db'))
-
+
+
+
+
+
+
-        cursor = cnx.cursor()
-
+
+
+
+
+
+
+
+
+
+
+
+
+
-        cnx.commit()
-
-    def insert_default_data(self):
@@ -138 +223 @@
-            if v == None:
-                return 'NULL'
-            else:
-
+
+
+
+
-        cnx = self.get_db_cnx()
-        cursor = cnx.cursor()
-        
@@ -149 +237 @@
-                values = ','.join(map(prep_value, row))
-                sql = "INSERT INTO %s (%s) VALUES(%s);" % (table, cols, values)
-                cursor.execute(sql)
+        cnx.commit()
+
+    def setup_default_config(self):
-        for s,n,v in db_default.default_config:
-            if not self.cfg.has_section(s):
-                self.cfg.add_section(s)
-            self.cfg.set(s, n, v)
-        self.save_config()
-        cnx.commit()
-
-    def get_version(self):
-        cnx = self.get_db_cnx()
@@ -218 +307 @@
-        cursor = cnx.cursor()
-        cursor.execute('SELECT filename,description,type,size,time,author,ipnr '
-                       'FROM attachment '
-type, id
+(type, id,)
-        return cursor.fetchall()
-    
-    def get_attachments_hdf(self, cnx, type, id, hdf, prefix):
@@ -290 +379 @@
-
-    def backup(self, dest=None):
-        """Simple SQLite-specific backup. Copy the database file."""
-
+
+
-        if db_str[:7] != 'sqlite:':
-
+
+
+
+
+
-        db_name = os.path.join(self.path, db_str[7:])
-        if not dest:
-            dest = '%s.%i.bak' % (db_name, self.get_version())
@@ -321 +415 @@
-                    err = 'No upgrade module for version %i (%s.py)' % (i, upg)
-                    raise EnvironmentError, err
-                d.do_upgrade(self, i, cursor)
-i
+s
-                           "name='database_version'", db_default.db_version)
-            self.log.info('Upgraded db version from %d to %d',
-                          dbver, db_default.db_version)
-            cnx.commit()
+            print 'Upgrade succeeded.'
-            return 1

trac/Ticket.py

@@ -73 +73 @@
-            self[Ticket.std_fields[i]] = row[i] or ''
-
-        cursor = db.cursor ()
-i
+s
-        rows = cursor.fetchall()
-        if rows:
-            for r in rows:
@@ -93 +93 @@
-            if not dict.has_key(name):
-                self[name] = '0'
-
-
+, dbms
-        """Add ticket to database"""
-        cursor = db.cursor()
-        assert not self.has_key('id')
@@ -106 +106 @@
-        std_fields = filter(lambda n: n[:7] != 'custom_', self.keys())
-        custom_fields = filter(lambda n: n[:7] == 'custom_', self.keys())
-        std_values = map(lambda n, self=self: self[n], std_fields)
+        
-        nstr = string.join(std_fields, ',')
-        vstr = ('%s,' * len(std_fields))[:-1]
-
+ 
-                       *std_values)
-
+
+
+
+
+
+
+
+
-        for name in custom_fields:
-            cursor.execute('INSERT INTO ticket_custom(ticket,name,value)'
-d
+s
-        db.commit()
-        self['id'] = id
-        self._forget_changes()
@@ -146 +154 @@
-        for name in self._old.keys():
-            if name[:7] == 'custom_':
-                fname = name[7:]
-
+
+
+
+
+
+
+
+
+
+
-                               ' VALUES(%s, %s, %s)', id, fname, self[name])
-            else:
-                fname = name
-cursor.execute ('UPDATE ticket SET %s=%s WHERE id=%s',
-          
+sql = "UPDATE ticket SET %s='%s' WHERE id=%s" \
+% (
-
+                cursor.execute (sql)
+
-            cursor.execute ('INSERT INTO ticket_change '
-                            '(ticket, time, author, field, oldvalue, newvalue) '
-                            'VALUES (%s, %s, %s, %s, %s, %s)',
@@ -171 +190 @@
-        """Returns the changelog as a list of dictionaries"""
-        cursor = db.cursor()
-        if when:
-'SELECT time, author, field, oldvalue, newvalue '
-'FROM ticket_change '
-'WHERE ticket=%s AND time=%s'
-'UNION '
-'SELECT time, author, "attachment", null, filename '
-'FROM attachment '
-'WHERE id=%s AND time=%s '
-'ORDER BY time'
+"SELECT time, author, field, oldvalue, newvalue "
+"FROM ticket_change "
+"WHERE ticket=%s AND time=%s"
+"UNION "
+"SELECT time, author, 'attachment', null, filename "
+"FROM attachment "
+"WHERE id=%s AND time=%s "
+"ORDER BY time"
-        else:
-'SELECT time, author, field, oldvalue, newvalue '
-'FROM ticket_change '
-'WHERE ticket=%s '
-'UNION '
-'SELECT time, author, "attachment", null,filename '
-'FROM attachment '
-'WHERE id = %s '
-'ORDER BY time'
+"SELECT time, author, field, oldvalue, newvalue "
+"FROM ticket_change "
+"WHERE ticket=%s "
+"UNION "
+"SELECT time, author, 'attachment', null,filename "
+"FROM attachment "
+"WHERE id = %s "
+"ORDER BY time"
-        log = []
-        while 1:
-            row = cursor.fetchone()
@@ -283 +302 @@
-            owner = cursor.fetchone()[0]
-            ticket['owner'] = owner
-
-
+
+
-
-        # Notify
-        tn = TicketNotifyEmail(self.env)

trac/WikiFormatter.py

@@ -130 +130 @@
-    def _changesethref_formatter(self, match, fullmatch):
-        number = int(match[1:-1])
-        cursor = self.db.cursor ()
-d
+s
-        row = cursor.fetchone ()
-        if not row:
-            return '[<a class="missing" href="%s">%d</a>]' % (self._href.changeset(number), number)

contrib/tracdb-sqlite2pg.py

@@ +1 @@
+#!/usr/bin/env python
+
+"""
+Import a Trac database from sqlite to postgresql.
+
+Requires:  Trac from http://trac.edgewall.com/
+           Python 2.3 from http://www.python.org/
+           PostgreSQL from http://www.postgresql.org/
+           PySQLITE ?
+           others?
+           
+Author: Brad Anderson <brad@dsource.org>
+
+
+"""
+
+import sys, os
+import sqlite
+import pgdb
+from trac.util import sql_escape
+
+tables=['revision','node_change', 'auth_cookie', 'enum', 'system', 'lock', 
+        'ticket', 'ticket_change', 'ticket_custom', 'report', 'permission',
+        'component', 'milestone', 'version', 'attachment', 'session', 'wiki',
+        'forums', 'topics', 'posts']
+
+# for debugging, one table at a time
+# tables=['wiki']
+
+
+## simple field translation mapping.  if string not in
+## mapping, just return string, otherwise return value
+#class FieldTranslator(dict):
+#    def __getitem__(self, item):
+#        if not dict.has_key(self, item):
+#            return item
+#            
+#        return dict.__getitem__(self, item)
+
+
+
+def convert(_env, _host, _db, _user, _password, _clean):
+    
+    
+    # init PostgreSQL environment
+    print "PostgreSQL('%s':'%s':'%s'): connecting..." \
+                      % (_host, _db, _user)
+    pg_con = pgdb.connect(host=_host, database=_db, 
+                          user=_user, password=_password)
+    pg_cur = pg_con.cursor()
+
+    # init Trac environment
+    print "Trac SQLite('%s'): connecting..." % (_env)
+    trac_con = sqlite.connect(os.path.join(_env, "db/trac.db"),
+                              timeout=10000)
+    trac_cur = trac_con.cursor()
+    print
+
+#    fieldtypes = FieldTranslator()
+#    fieldtypes['0'] = 'int'
+#    fieldtypes['6'] = 'text'
+    
+    # loop thru tables, converting them
+    for t in tables:
+        print
+        print "Converting Table: '%s'" % t
+
+        # clear out existing table
+#        if _clean == 1:
+#            pg_cur.execute("truncate table %s;" % t)
+        pg_cur.execute("truncate table %s;" % t)
+        
+        # get table info
+        sql = "SELECT * FROM %s" % t
+        trac_cur.execute(sql)
+        cols = trac_cur.rs.col_defs
+        counter = 0
+        
+        # loop thru rows, moving data from sqlite to pgsql
+        while 1:
+            row = trac_cur.fetchone()
+            counter += 1
+            if not row:
+                break
+            flds=""
+            vals=""
+            for c in cols:
+                field = c[0]
+                type = c[1]
+                value = row[field]
+                if value != None:
+                    flds += ", %s" % field
+                    if type in [6]:
+                        value = "'%s'" % sql_escape(value)
+                    vals += ", %s" % value
+            flds = flds[2:]
+            vals = vals[2:]
+            sql  = "INSERT INTO %s " % t
+            sql += "(%s) " % flds
+            sql += "VALUES (%s);" % vals
+
+            pg_cur.execute(sql)
+            pg_con.commit()
+            
+            if counter % 100 == 0:
+                print "%s rows" % counter
+        
+    trac_cur.close()
+    trac_con.close()
+    pg_cur.close()
+    pg_con.close()
+    print
+        
+    
+def usage():
+    print "tracdb-sqlite2pg - Converts a Trac database from sqlite to " \
+          "postgresql."
+    print
+    print "This script is designed to transfer an existing Trac SQLite "
+    print "database to an existing Trac PosgreSQL database.  The two " 
+    print "databases should be of the same version.  Check in the 'system' "
+    print "table in each database to verify.  This script runs on version 8 "
+    print "and above.  This is for Trac pre 0.9 (revision 1169 and higher)."
+    print
+    print "Note: you can make a Trac PosgreSQL database with trac-admin "
+    print "(the version that accompanies this script)"
+    print
+    print "THIS WILL DESTROY ALL DATA IN THE POSTGRESQL DATABASE !! "
+    print
+    print
+    print "Usage: tracdb-sqlite2pg.py [options]"
+    print
+    print "Available Options:"
+    print "  -t  | --tracenv /path/to/trac/env - full path to Trac environment"
+    print "  -h  | --host <pgsql hostname>     - PostgreSQL hostname"
+    print "  -db | --database <pgsql database> - PostgreSQL database name"
+    print "  -u  | --user <pgsql username>     - PostgreSQL username"
+    print "  -p  | --passwd <pgsql password>   - PostgreSQL password"
+#    print "  -c  | --clean                     - clean out PG database " \
+#          "before converting. (recommended)"
+    print "  --help | help                     - this help info"
+    print
+    print "Additional configuration options can be defined directly in the " \
+          "script."
+    print
+    sys.exit(0)
+
+def main():
+    global TRAC_ENV, PG_HOST, PG_DB, PG_USER, PG_PASSWORD, PG_CLEAN
+    if len (sys.argv) > 1:
+        if sys.argv[1] in ['--help','help'] or len(sys.argv) < 4:
+            usage()
+        iter = 1
+        while iter < len(sys.argv):
+            if sys.argv[iter] in ['-t', '--tracenv'] and iter+1 < len(sys.argv):
+                TRAC_ENV = sys.argv[iter+1]
+                iter = iter + 1
+            elif sys.argv[iter] in ['-h', '--host'] and iter+1 < len(sys.argv):
+                PG_HOST = sys.argv[iter+1]
+                iter = iter + 1
+            elif sys.argv[iter] in ['-db', '--database'] and \
+                    iter+1 < len(sys.argv):
+                PG_DB = sys.argv[iter+1]
+                iter = iter + 1
+            elif sys.argv[iter] in ['-u', '--user'] and iter+1 < len(sys.argv):
+                PG_USER = sys.argv[iter+1]
+                iter = iter + 1
+            elif sys.argv[iter] in ['-p', '--passwd'] and \
+                    iter+1 < len(sys.argv):
+                PG_PASSWORD = sys.argv[iter+1]
+                iter = iter + 1
+#            elif sys.argv[iter] in ['-c', '--clean']:
+#                PG_CLEAN = 1
+            else:
+                print "Error: unknown parameter: " + sys.argv[iter]
+                sys.exit(0)
+            iter = iter + 1
+    else:
+        usage()
+        
+    convert(TRAC_ENV, PG_HOST, PG_DB, PG_USER, PG_PASSWORD, PG_CLEAN)
+
+
+if __name__ == '__main__':
+    main()