Source

htsql-firebird / src / htsql_pgsql / tweak / view / introspect.py

#
# Copyright (c) 2006-2011, Prometheus Research, LLC
# See `LICENSE` for license information, `AUTHORS` for the list of authors.
#


from htsql_pgsql.core.introspect import IntrospectPGSQL
from htsql.core.entity import PrimaryKeyEntity, ForeignKeyEntity, CatalogEntity
from . import rulesparser


class PGViewIntrospectPGSQL(IntrospectPGSQL):

    def introspect_catalog(self):
        schemas = self.introspect_schemas()
        self.introspect_views()
        return CatalogEntity(schemas)

    def introspect_views(self):
        for oid in self.meta.pg_rewrite:
            rule = self.meta.pg_rewrite[oid]
            if rule.ev_type != '1' \
                    or rule.ev_attr >= 0 \
                    or not rule.is_instead \
                    or rule.ev_qual != '<>':
                # not a view
                continue

            if not rule.ev_class in self.table_by_oid:
                # not introspected view
                continue

            view = self.table_by_oid[rule.ev_class]

            ruletree = rulesparser.RuleTreeParser().parse(rule.ev_action)
            for scenario in rulesparser.scenario_list:
                if scenario.accepts(ruletree):
                    keyset = scenario.find_keys(ruletree, view, self.table_by_oid)
                    for key in keyset:
                        if isinstance(key, PrimaryKeyEntity):
                            view.unique_keys.append(key)
                            view.primary_key = key
                        if isinstance(key, ForeignKeyEntity):
                            view.foreign_keys.append(key)
                    if len(keyset) > 0:
                        break
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.