Let quote_id do the dot-join of quoted names

The '%s.%s' % (quote_id(foo), quote_id(bar)) is used inconsistently.

pyrseas/augment/table.py

@@ -70,7 +70,7 @@ def link_current(self, tables):
         """
         for (sch, tbl) in self:
             if not (sch, tbl) in tables:
-                raise KeyError("Table %s.%s not in current database" % (
-                    sch, tbl))
+                raise KeyError("Table %s not in current database" %
+                    quote_id(sch, tbl))
             if not hasattr(self[(sch, tbl)], 'current'):
                 self[(sch, tbl)].current = tables[(sch, tbl)]

pyrseas/augment/trigger.py

@@ -7,7 +7,7 @@
     DbAugment and CfgTriggerDict derived from DbAugmentDict.
 """
 from pyrseas.augment import DbAugmentDict, DbAugment
-from pyrseas.dbobject import split_schema_obj
+from pyrseas.dbobject import split_schema_obj, quote_id
 from pyrseas.dbobject.trigger import Trigger
 
 
@@ -35,7 +35,7 @@ def apply(self, table):
                     newtrg.procedure[:14], table.name)
             (sch, fnc) = split_schema_obj(newtrg.procedure)
             if sch != table.schema:
-                newtrg.procedure = "%s.%s" % (table.schema, fnc)
+                newtrg.procedure = quote_id(table.schema, fnc))
         table.triggers.update({newtrg.name: newtrg})
 
 

pyrseas/dbobject/__init__.py

@@ -39,23 +39,28 @@ def fetch_reserved_words(db):
                                          WHERE catcode = 'R'""")]
 
 
-def quote_id(name):
+def quote_id(*names):
     """Quotes an identifier if necessary.
 
-    :param name: string to be quoted
+    :param names: strings to be quoted. If more than one they will be merged
+        by dot.
 
     :return: possibly quoted string
     """
-    regular_id = True
-    if not name[0] in VALID_FIRST_CHARS or name in RESERVED_WORDS:
-        regular_id = False
-    else:
-        for ltr in name[1:]:
-            if ltr not in VALID_CHARS:
-                regular_id = False
-                break
+    rv = []
+    for name in names:
+        regular_id = True
+        if not name[0] in VALID_FIRST_CHARS or name in RESERVED_WORDS:
+            regular_id = False
+        else:
+            for ltr in name[1:]:
+                if ltr not in VALID_CHARS:
+                    regular_id = False
+                    break
+
+        rv.append(regular_id and name or '"%s"' % name.replace('"', '""'))
 
-    return regular_id and name or '"%s"' % name
+    return '.'.join(rv)
 
 
 def split_schema_obj(obj, sch=None):
@@ -603,7 +608,7 @@ def qualname(self, objname=None):
         if objname is None:
             objname = self.name
         return self.schema == 'public' and quote_id(objname) \
-            or "%s.%s" % (quote_id(self.schema), quote_id(objname))
+            or quote_id(self.schema, objname)
 
     def unqualify(self):
         """Adjust the schema and table name if the latter is qualified"""
@@ -619,8 +624,8 @@ def extern_filename(self, ext='yaml'):
         return super(DbSchemaObject, self).extern_filename(ext, True)
 
     def rename(self, oldname):
-        return "ALTER %s %s.%s RENAME TO %s" % (
-            self.objtype, quote_id(self.schema), quote_id(oldname),
+        return "ALTER %s %s RENAME TO %s" % (
+            self.objtype, quote_id(self.schema, oldname),
             quote_id(self.name))
 
     def get_implied_deps(self, db):

pyrseas/dbobject/constraint.py

@@ -84,8 +84,8 @@ def get_implied_deps(self, db):
         elif isinstance(self._table, Domain):
             deps.add(db.types[self.schema, self.table])
         else:
-            raise KeyError("Constraint '%s.%s' on unknown type/class" % (
-                self.schema, self.name))
+            raise KeyError("Constraint '%s' on unknown type/class" % (
+                quote_id(self.schema, self.name)))
 
         return deps
 
@@ -471,7 +471,7 @@ def _from_catalog(self):
             constr.unqualify()
             oid = constr.oid
             sch, tbl, cns = constr.key()
-            sch, tbl = split_schema_obj('%s.%s' % (sch, tbl))
+            sch, tbl = split_schema_obj(quote_id(sch, tbl))     # TODO why?
             constr_type = constr.type
             del constr.type
             if constr_type != 'f':

pyrseas/dbobject/index.py

@@ -216,7 +216,7 @@ def _from_catalog(self):
             index.unqualify()
             oid = index.oid
             sch, tbl, idx = index.key()
-            sch, tbl = split_schema_obj('%s.%s' % (sch, tbl))
+            sch, tbl = split_schema_obj(quote_id(sch, tbl))     # TODO: why?
             keydefs, _, _ = index.defn.partition(' WHERE ')
             _, _, keydefs = keydefs.partition(' USING ')
             keydefs = keydefs[keydefs.find(' (') + 2:-1]

pyrseas/dbobject/operator.py

@@ -68,8 +68,8 @@ def qualname(self):
 
         No qualification is used if the schema is 'public'.
         """
-        return self.schema == 'public' and self.name \
-            or "%s.%s" % (quote_id(self.schema), self.name)
+        return self.schema == 'public' and quote_id(self.name) \
+            or quote_id(self.schema, self.name)
 
     def identifier(self):
         """Return a full identifier for an operator object

pyrseas/dbobject/table.py

@@ -58,7 +58,7 @@ def get_attrs(self, dbconn):
         data = dbconn.fetchone(
             """SELECT start_value, increment_by, max_value, min_value,
                       cache_value
-               FROM %s.%s""" % (quote_id(self.schema), quote_id(self.name)))
+               FROM %s""" % quote_id(self.schema, self.name))
         for key, val in list(data.items()):
             setattr(self, key, val)
 

pyrseas/relation/join.py

@@ -2,6 +2,7 @@
 """
     pyrseas.relation.join
 """
+from pyrseas.dbobject import quote_id
 from pyrseas.relation.attribute import Attribute
 from pyrseas.relation.tuple import Tuple
 
@@ -98,9 +99,9 @@ def where_clause(self, qry_args=None):
         attrs = {}
         for name, attr in self.attributes:
             if attr.name != attr.basename:
-                expr = "%s.%s" % (attr.projection.rangevar, attr.basename)
+                expr = quote_id(attr.projection.rangevar, attr.basename)
             else:
-                expr = "%s.%s" % (attr.projection.rangevar, attr.name)
+                expr = quote_id(attr.projection.rangevar, attr.name)
             attrs.update({attr.name: (expr, attr.type)})
         subclauses = []
         params = {}
@@ -156,12 +157,12 @@ def getsubset_qry():
                 exprs = []
                 for name, attr in self.attributes:
                     if attr.name != attr.basename:
-                        exprs.append("%s.%s AS %s" % (
-                            attr.projection.rangevar, attr.basename,
-                            attr.name))
+                        exprs.append("%s AS %s" % (
+                            quote_id(attr.projection.rangevar, attr.basename),
+                            quote_id(attr.name)))
                     else:
-                        exprs.append("%s.%s" % (attr.projection.rangevar,
-                                                attr.name))
+                        exprs.append(
+                            quote_id(attr.projection.rangevar, attr.name))
                 self.getsubset_qry = "SELECT %s FROM %s" % (
                     ", ".join(exprs), self.from_clause)
             return self.getsubset_qry