pg_meta 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b58a4f531bb7de0145b96bad4e39f8c7357d14e300c0d91b7fc5af44567974e4
-  data.tar.gz: 6e5144a8afc064666199f1c692e1eb8f3760ac7688f3d58a1ed168dcd4ab7ac4
+  metadata.gz: 2862a980ca3d5715c3d60daacd06e78d4b0ae1baccb22e78d5cc97441d303839
+  data.tar.gz: 3619f169ae0641575b8b0e268f2db4f1f1da0655b88f19b0ba4767e814004416
 SHA512:
-  metadata.gz: 5f3c8c0c82a7099858decef204aaeb51116a2980ccf96960424d70864df0b8bd8e9f14269740bf94ec5aec5e9a4254def92b49bf7217599c83c712091e85d66c
-  data.tar.gz: 4d8237a0bf64240a222541b917f6c017afef72c863e3c18be20427060554d6fee903779c2c8f24cde23bde8b996b9ed0b51f8c45d7891ac0ae9c6245640c28c5
+  metadata.gz: 671e634a25906f96b1b723ae1e92d2d531256c0cd3c35b9f8b5dda98aed755f16816950e50df05d0dc13ab9ff52bc30182b7e0077ea37b73ec47df322774c764
+  data.tar.gz: 9d767775656d5515e9d38b63e4e6234a806a2f90947a5916b3074d26c6510ddbd69116f8682144f1bb1a83fff8146e5acdffa3bcc9b7909afda9c8bc44a983c3

data/exe/pg_meta

@@ -6,6 +6,9 @@ require 'pg_meta'
 include ShellOpts
 
 SPEC = %(
+  +e,exclude=SCHEMA
+    Exclude SCHEMA. This option can be repeated
+
   -f,format=FORMAT:yaml,marshal,dump
     Control format of output. FORMAT can be one of 'yaml', 'marshal' or the
     human-readable 'dump' format (the default)
@@ -17,7 +20,7 @@ opts, args = ShellOpts.process(SPEC, ARGV, exception: false)
 database = args.expect(1)
 
 conn = PgConn.new(database)
-meta = PgMeta.new(conn)
+meta = PgMeta.new(conn, exclude_schemas: opts.exclude)
 
 case opts.format
   when "yaml"; puts meta.to_yaml

data/lib/pg_meta/load_conn.rb

@@ -5,16 +5,25 @@ module PgMeta
 
     # Load the database from the given PgConn::Connection object. This method is a
     # transaction wrapper around #do_load_conn
-    def self.load_conn(pg_conn)
+    def self.load_conn(pg_conn, exclude_schemas: [])
       pg_conn.pg_connection.transaction { |conn| 
         conn.exec "set transaction isolation level serializable read only deferrable"
-        do_load_from_connection(conn) 
+        do_load_from_connection(conn, exclude_schemas: exclude_schemas) 
       }
     end
 
   private
+    # Generate SQL to exclude schemas
+    def self.exclude_sql(field, exclude_schemas)
+      sql = "#{field} !~ '^pg_' and #{field} <> 'information_schema'"
+      if !exclude_schemas.empty?
+        sql += " and #{field} not in ('#{exclude_schemas.join("', '")}')"
+      end
+      sql
+    end
+
     # Load the given database into self. Note that +conn+ is a PG::Connection object
-    def self.do_load_from_connection(conn)
+    def self.do_load_from_connection(conn, exclude_schemas: [])
       # Get database owner
       r = conn.exec %(
         select  rolname::varchar
@@ -27,12 +36,15 @@ module PgMeta
       # Create database object
       db = Database.new(conn.db, owner)
 
+      # Schema excludes
+      exclude_sql = 
+
       # Build schemas
       conn.exec(%(
         select  schema_name::text as name,
                 schema_owner::text as owner
         from    information_schema.schemata
-        where   schema_name !~ '^pg_' and schema_name <> 'information_schema' 
+        where   #{exclude_sql("schema_name", exclude_schemas)}
       )).each { |row|
         Schema.new(db, *row.project(:name, :owner))
       }
@@ -49,7 +61,7 @@ module PgMeta
         left join information_schema.tables i on
                     i.table_schema = n.nspname
                 and i.table_name = c.relname
-        where   n.nspname !~ '^pg_' and n.nspname <> 'information_schema'
+        where   #{exclude_sql("n.nspname", exclude_schemas)}
         and     c.relkind in ('r', 'v', 'm')
       )).each { |row|
         schema = db.schemas[row[:schema]]
@@ -111,7 +123,7 @@ module PgMeta
         join    pg_rewrite r on r.oid = d.objid
         join    pg_class c2 on c2.oid = r.ev_class
         join    pg_namespace n2 on n2.oid = c2.relnamespace
-        where   n.nspname !~ '^pg_' and n.nspname <> 'information_schema'
+        where   #{exclude_sql("n.nspname", exclude_schemas)}
         and     c.relkind in ('r', 'v', 'm')
         and     d.classid = 'pg_rewrite'::regclass
         and     d.refclassid = 'pg_class'::regclass
@@ -149,7 +161,7 @@ module PgMeta
                     i.table_schema = n.nspname
                 and i.table_name = c.relname
                 and i.column_name = a.attname
-        where   n.nspname !~ '^pg_' and n.nspname <> 'information_schema'
+        where   #{exclude_sql("n.nspname", exclude_schemas)}
         and     c.relkind in ('r', 'v', 'm')
         and     a.attnum > 0
       )).each { |row|
@@ -178,7 +190,7 @@ module PgMeta
         left join information_schema.check_constraints cc
                   on  cc.constraint_schema = c.table_schema and
                       cc.constraint_name = c.constraint_name
-        where   c.table_schema !~ '^pg_' and c.table_schema <> 'information_schema'
+        where   #{exclude_sql("c.table_schema", exclude_schemas)}
         and     c.constraint_type in ('PRIMARY KEY', 'UNIQUE', 'CHECK')
       )).each { |row|
         table = db.schemas[row[:schema]].tables[row[:table]]
@@ -212,7 +224,7 @@ module PgMeta
         join    pg_index x on x.indrelid = c.oid
         join    pg_class i on i.oid = x.indexrelid
         join    ( select attrelid, attname from pg_attribute order by attnum ) a on a.attrelid = i.oid
-        where   n.nspname !~ '^pg_' and n.nspname <> 'information_schema'
+        where   #{exclude_sql("n.nspname", exclude_schemas)}
         and     i.oid not in (select conindid from pg_constraint)
         and     x.indisunique
         group by
@@ -261,7 +273,7 @@ module PgMeta
         join    information_schema.key_column_usage cu_refed 
                 on  cu_refed.constraint_schema = rc.unique_constraint_schema
                     and cu_refed.constraint_name = rc.unique_constraint_name
-        where   cu_refing.table_schema !~ '^pg_' and cu_refing.table_schema <> 'information_schema'
+        where   #{exclude_sql("cu_refing.table_schema", exclude_schemas)}
         group by
                 rc.constraint_schema,
                 rc.constraint_name,
@@ -300,7 +312,7 @@ module PgMeta
                 end as "security"
         from    pg_proc p
         join    pg_namespace s on (p.pronamespace = s.oid)
-        where   s.nspname !~ '^pg_' and s.nspname <> 'information_schema'
+        where   #{exclude_sql("s.nspname", exclude_schemas)}
       )).each { |row|
         schema = db.schemas[row[:schema]]
         klass = (row[:kind] == 'function' ? Function : Procedure)

data/lib/pg_meta/version.rb

@@ -1,3 +1,3 @@
 module PgMeta
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

data/lib/pg_meta.rb

@@ -19,8 +19,8 @@ module PgMeta
   #
   # Initialize a Database object
   #
-  def self.new(*args)
-    Database.load_conn(PgConn.ensure(*args))
+  def self.new(*args, exclude_schemas: [])
+    Database.load_conn(PgConn.ensure(*args), exclude_schemas: exclude_schemas)
   end
 
   # Load data from a YAML object

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_meta
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Claus Rasmussen
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-04-24 00:00:00.000000000 Z
+date: 2022-05-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: indented_io