pg_meta 0.1.2 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8553533c4f97ee625c2796b354d0ab5124786853f3a64f1b1a6b0d126ba8ac35
-  data.tar.gz: 5e80ab9a799e73c21061fc78275affe48706208246b749a198223f7b8f8a0dbf
+  metadata.gz: 06b01cd5716ff1dbc04209a9b7792f59808b8d23f633b1810528123d62b19540
+  data.tar.gz: 2b6faf8696583d6c46893c6454dd114d9b83aedb452fa636560da255598d3f24
 SHA512:
-  metadata.gz: 80ad81f85a6acb805971f5522cd3189ba553c0f4ab0a6d09c9f152084ac099d3aff1399da10bc71222f18551e3ac2ed9b2b28ea7b9304a233c943cfbde9dac0e
-  data.tar.gz: 548de1800985a70250558ce8611ca08f81c472caa5f0d9da5345ace172d5ef37103e1fa6475325b47eb6b9dc83719e9f5c65495610e15d88916717d3dfd37f46
+  metadata.gz: bc2a8a658757b86f89926bac1654c27a8ef00b4e3f7028b1c65726f684e8cbb58ed86c2b87ca540b5fb48b2739fdc15b396fe7cc8fb58ec720f252ba95b2de97
+  data.tar.gz: 7bf6217586f5d96e3ec7c6471da2472c36c0beaadf7d8f114d51adb242bd473af9bf559e525e354531580fdc6b5ecf501be5e8087a18986ee82177abf59fca55

data/.ruby-version

@@ -1 +1 @@
-ruby-2.7.1
+ruby-3.1.2

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)
@@ -13,13 +16,15 @@ SPEC = %(
   -- DATABASE
 )
 
-opts, args = ShellOpts.process(SPEC, ARGV, version: PgMeta::VERSION, exception: false)
+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
   when "marshal"; puts meta.to_marshal
   else meta.dump
 end
+

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.2"
+  VERSION = "0.2.0"
 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

data/pg_meta.gemspec

@@ -25,7 +25,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency "pg"
 
   spec.add_dependency "pg_conn"
-  spec.add_dependency "shellopts", "~> 2.0.0"
+  spec.add_dependency "shellopts", "~> 2.1"
 
   spec.add_development_dependency "bundler"
   spec.add_development_dependency "rake"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_meta
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.0
 platform: ruby
 authors:
 - Claus Rasmussen
-autorequire: 
+autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-04-24 00:00:00.000000000 Z
+date: 2022-07-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: indented_io
@@ -58,14 +58,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.0.0
+        version: '2.1'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.0.0
+        version: '2.1'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -152,7 +152,7 @@ files:
 homepage: http://www.nowhere.com/
 licenses: []
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -167,8 +167,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.4
-signing_key: 
+rubygems_version: 3.3.18
+signing_key:
 specification_version: 4
 summary: pg_meta gem
 test_files: []