Changeset 155 for trunk/module/pg.py


Ignore:
Timestamp:
Feb 20, 2004, 8:22:47 AM (16 years ago)
Author:
darcy
Message:

Add interval type.

Use list comprehensions for efficiency.

File:
1 edited

Legend:

Unmodified
Added
Removed
  • trunk/module/pg.py

    r152 r155  
    117117
    118118        def get_databases(self):
    119                 l = []
    120                 for n in self.db.query("SELECT datname FROM pg_database").getresult():
    121                         l.append(n[0])
    122                 return l
     119                return [x[0] for x in
     120                        self.db.query("SELECT datname FROM pg_database").getresult()]
    123121
    124122        def get_tables(self):
    125                 l = []
    126                 for n in self.db.query("""SELECT relname FROM pg_class
     123                return [x[0] for x in
     124                                self.db.query("""SELECT relname FROM pg_class
    127125                                                WHERE relkind = 'r' AND
    128126                                                        relname !~ '^Inv' AND
    129                                                         relname !~ '^pg_'""").getresult():
    130                         l.append(n[0])
    131                 return l
     127                                                        relname !~ '^pg_'""").getresult()]
    132128
    133129        def get_attnames(self, cl, newattnames = None):
     
    156152                l = {}
    157153                for attname, typname in self.db.query(query % cl).getresult():
     154                        if re.match("^interval", typname):
     155                                l[attname] = 'text'
    158156                        if re.match("^int", typname):
    159157                                l[attname] = 'int'
Note: See TracChangeset for help on using the changeset viewer.