database-model-generator 0.6.1 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6be4d64e5f85a52ac4fd634f0a11f80c6ea8294065da9d258dd5e35084d31fd9
-  data.tar.gz: f1fc9410a6584615eac0d0bc11a298926eef3606589bb0bc48c624e80667ea75
+  metadata.gz: f565ac034c268e0422b5da72e50d74cd527b14537a903999cb540a6d61cade81
+  data.tar.gz: c6472df0598d9118ec24ae8e9d44c81610b7c3aa2e02504054dcbdfffe1e3007
 SHA512:
-  metadata.gz: dffec8aa82468f6723f6c9984f7ff9b87328cdf5fe506380437f782d890300571983cdfe02be7fd5d64ec9d16e615df9813e8b3087704b5d75e8a7fe1aadcc1a
-  data.tar.gz: d5a64c260219b5010e312eb4604c8b09005cf2fd6e63325e5741e652c04cd1a29e944ecb42a35f314632b5571b6997a4982a309d3b30adb38b7eb9b28d2ed776
+  metadata.gz: e71a3140a6c228bd2e0bba38e2ce3fb316d1e864a5823afb673d5dc4aec4d6eb6a3f6fee87e6861faf3cdddc59cab372c15bfcd18b736ad77662dce34d2e5648
+  data.tar.gz: '09cf787750a59cff756ac3a2fae2f607fe6125ef5fcbb59dfe9395f4d79012fa29aa07093796910fe753854f82d084f47379cff22388f3e383719d989e1eb4ee'

data/CHANGES.md

@@ -1,3 +1,11 @@
+## 0.7.0 - 22-Dec-2025
+* Added full PostgreSQL support via the pg gem.
+* Created PostgreSQL generator implementation with all features.
+* Added PostgreSQL Docker setup for testing.
+* Added comprehensive PostgreSQL test suite.
+* Updated documentation to include PostgreSQL examples.
+* Added pg gem as development dependency.
+
 ## 0.6.1 - 28-Jul-2025
 * Added "datetime" field for validates_timeliness.
 * Added validates_timeliness as a development dependency.

data/MANIFEST.md

@@ -1,10 +1,38 @@
 CHANGES.md
 MANIFEST.md
 README.md
-oracle-model-generator.gemspec
+LICENSE
+database-model-generator.gemspec
+Gemfile
 Rakefile
-bin/omg
+bin/dmg
 certs/djberg96_pub.pem
-lib/oracle-model-generator.rb
+lib/database_model_generator.rb
 lib/oracle/model/generator.rb
-test/test_oracle_model_generator.rb
+lib/sqlserver/model/generator.rb
+lib/postgresql/model/generator.rb
+spec/spec_helper.rb
+spec/oracle_model_generator_spec.rb
+spec/postgresql_model_generator_spec.rb
+spec/support/oracle_connection.rb
+spec/support/postgresql_connection.rb
+docker/oracle/docker-compose.yml
+docker/oracle/Dockerfile
+docker/oracle/README.md
+docker/oracle/test.sh
+docker/sqlserver/docker-compose.yml
+docker/sqlserver/Dockerfile
+docker/sqlserver/test-Dockerfile
+docker/sqlserver/init-db.sql
+docker/sqlserver/setup-db.sh
+docker/sqlserver/run_tests.sh
+docker/sqlserver/test.sh
+docker/sqlserver/DOCKER.md
+docker/sqlserver/SUPPORT.md
+docker/sqlserver/TESTING.md
+docker/postgresql/docker-compose.yml
+docker/postgresql/Dockerfile
+docker/postgresql/test-Dockerfile
+docker/postgresql/init-db.sql
+docker/postgresql/test.sh
+docker/postgresql/README.md

data/README.md

@@ -1,7 +1,7 @@
 ## Description
 A library for generating an ActiveRecord model from an existing database table.
 
-Currently supports both Oracle and SQL Server databases.
+Currently supports Oracle, SQL Server, and PostgreSQL databases.
 
 This will install a "dmg" (Database Model Generator) executable that you can
 use from the command line.
@@ -9,7 +9,7 @@ use from the command line.
 ## Renamed
 Originally called "oracle-model-generator" and put into the dust bin, I've
 decided to revive this library with the help of AI. Specifically, I've added
-SQLServer support, and plan to add Postgres support.
+SQLServer and PostgreSQL support, and plan to add more features.
 
 I also plan on lots of improvements, and some general refactoring.
 
@@ -22,6 +22,9 @@ Using the command line tool:
 ### SQL Server:
 `dmg -T sqlserver -s localhost -d your_database -t locations -u sa -p your_password`
 
+### PostgreSQL:
+`dmg -T postgresql -h localhost -d your_database -t locations -u postgres [-p password]`
+
 ### Auto-detection:
 # Oracle (default)
 `dmg -d your_database -t locations -u some_user -p some_password`
@@ -124,6 +127,9 @@ end
 ### SQLServer
 * tiny_tds
 
+### PostgreSQL
+* pg
+
 ## Running the specs
 ### Oracle:
 Run `cd docker/oracle && docker-compose run --rm oracle-model-generator bundle exec rspec`.
@@ -134,6 +140,10 @@ with database client libraries.
 ### SQL Server:
 Run `cd docker/sqlserver && ./test.sh` to start SQL Server, then run tests.
 
+### PostgreSQL:
+Run `cd docker/postgresql && docker-compose up -d postgres-db` to start PostgreSQL,
+then run `docker-compose run --rm dmg_test` to execute tests.
+
 Again, no guarantees on MacOS.
 
 ## Optional Libraries
@@ -149,19 +159,23 @@ If you want date format validations, then you will need to install the
 ## Database Support
 * **Oracle**: Full support via ruby-oci8
 * **SQL Server**: Full support via tiny_tds
+* **PostgreSQL**: Full support via pg
 * **Auto-detection**: Automatically detects database type based on connection parameters
 
 ## Author's Comments
 Originally focused only on Oracle, this library has been expanded to support
-SQL Server as well. The architecture now supports multiple database vendors,
-and I will probably add Postgres support in the future.
+SQL Server and PostgreSQL as well. The architecture now supports multiple
+database vendors, making it easy to add support for additional databases in
+the future.
 
 ## Current Features
-* **Multi-database support**: Oracle and SQL Server
+* **Multi-database support**: Oracle, SQL Server, and PostgreSQL
 * **Auto-detection**: Automatically detects database type
 * **Index recommendations**: Suggests optimal indexes for your tables
 * **Multiple test frameworks**: Supports test-unit, minitest, and rspec
-* **Docker support**: Complete Docker environments for testing both databases
+* **Docker support**: Complete Docker environments for testing all databases
+* **Enum detection**: Automatically detects enum columns from CHECK constraints
+* **Polymorphic associations**: Identifies polymorphic relationship patterns
 
 ## Future Plans (originally)
 * Add support for views.

data/bin/dmg

@@ -40,9 +40,9 @@ def help
     -u, --user     => The username used to establish a connection to the database.
     -p, --password => The password used to establish a connection to the database.
     -d, --database => The name of the database to connect to.
-    -s, --server   => The database server hostname (for SQL Server).
-    -P, --port     => The database server port (default: 1433 for SQL Server).
-    -T, --type     => Database type: 'oracle' or 'sqlserver' (auto-detected if not specified).
+    -s, --server   => The database server hostname (for SQL Server and PostgreSQL).
+    -P, --port     => The database server port (default: 1433 for SQL Server, 5432 for PostgreSQL).
+    -T, --type     => Database type: 'oracle', 'sqlserver', or 'postgresql' (auto-detected if not specified).
     -r, --rails    => The version of rails you're using (2 or higher).
     -x, --tests    => Generate tests using testunit, minitest or rspec.
     -c, --class    => Class name for the generated table (optional)
@@ -55,13 +55,18 @@ def help
     SQL Server:
     dmg -T sqlserver -s localhost -P 1433 -d mydb -u sa -p password -t users
 
+    PostgreSQL (password optional):
+    dmg -T postgresql -s localhost -d mydb -u postgres -t users
+    dmg -T postgresql -s localhost -d mydb -u postgres -p password -t users
+
     Auto-detection (Oracle if no server specified, SQL Server if server specified):
     dmg -d localhost:1521/XE -u scott -p tiger -t users        # Oracle
     dmg -s localhost -d mydb -u sa -p password -t users        # SQL Server
 
-    If no user or password are supplied, then the generator will attempt to glean that
-    information using a combination of the database name and your .dbrc file.
-    If that cannot be found, then an error is raised.
+    If no user is supplied, an error is raised. For PostgreSQL, password is optional
+    if pg_hba.conf is configured for trust authentication. For Oracle and SQL Server,
+    if no password is supplied, the generator will attempt to find credentials in your
+    .dbrc file using the dbi-dbrc library.
 
     If no output file is supplied then the file generated will match the name
     of the table, minus the 's' if present, with a .rb extension. This is lazy,
@@ -110,6 +115,7 @@ pass = opts['password']
 
 # Determine database type
 database_type = opts['type']&.downcase
+
 unless database_type
   if opts['server'] || opts['port']
     database_type = 'sqlserver'
@@ -118,24 +124,31 @@ unless database_type
   end
 end
 
-unless ['oracle', 'sqlserver'].include?(database_type)
-  puts "Invalid database type: #{database_type}. Must be 'oracle' or 'sqlserver'."
+unless ['postgresql', 'oracle', 'sqlserver'].include?(database_type)
+  puts "Invalid database type: #{database_type}. Must be 'oracle', 'sqlserver', or 'postgresql'."
   exit!
 end
 
-unless user && pass
+unless user
+  puts "You must specify a user with the -u option."
+  exit!
+end
+
+# For PostgreSQL, password is optional (can use pg_hba.conf trust authentication)
+# For Oracle and SQL Server, password is required unless using dbrc
+unless pass || database_type == 'postgresql'
   begin
     dbrc = DBI::DBRC.new(opts['database'], user)
     user = dbrc.user
     pass = dbrc.passwd
   rescue NameError
-    msg = "If you do not specify a username or password on the command line "
+    msg = "If you do not specify a password on the command line "
     msg << "then you must use the dbi-dbrc library and create a .dbrc file in "
     msg << "your home directory."
     puts msg
     exit!
   rescue DBI::DBRC::Error
-    msg = "No user or password provided, and no dbrc entry found for '"
+    msg = "No password provided, and no dbrc entry found for '"
     msg << opts['database'] + "'."
     puts msg
     exit!
@@ -169,6 +182,20 @@ begin
       port: port.to_i,
       database: opts['database']
     )
+  when 'postgresql'
+    require 'pg'
+    server = opts['server'] || 'localhost'
+    port = opts['port'] || 5432
+
+    conn_params = {
+      host: server,
+      port: port.to_i,
+      dbname: opts['database'],
+      user: user
+    }
+    conn_params[:password] = pass if pass
+
+    connection = PG.connect(conn_params)
   end
 rescue LoadError => e
   case database_type
@@ -176,6 +203,8 @@ rescue LoadError => e
     puts "Oracle support requires the 'oci8' gem. Install with: gem install oci8"
   when 'sqlserver'
     puts "SQL Server support requires the 'tiny_tds' gem. Install with: gem install tiny_tds"
+  when 'postgresql'
+    puts "PostgreSQL support requires the 'pg' gem. Install with: gem install pg"
   end
   exit!
 rescue => e
@@ -312,6 +341,7 @@ File.open(ofile, 'w') do |fh|
   if rails == 2
     # Character fields, size
     omg.column_info.each{ |col|
+      next if omg.primary_keys.map(&:downcase).include?(col.name.downcase) # Skip primary keys
       data_type = col.data_type.to_s
       if ['char', 'varchar', 'varchar2'].include?(data_type)
         validation = "validates_size_of :#{col.name.downcase}, :maximum => #{col.data_size}"
@@ -324,6 +354,7 @@ File.open(ofile, 'w') do |fh|
 
     # Fields that must be present
     omg.column_info.each{ |col|
+      next if omg.primary_keys.map(&:downcase).include?(col.name.downcase) # Skip primary keys
       unless col.nullable?
         validation = "validates_presence_of :#{col.name.downcase}"
         fh.puts "  #{validation}"
@@ -334,6 +365,7 @@ File.open(ofile, 'w') do |fh|
 
     # Numeric fields
     omg.column_info.each{ |col|
+      next if omg.primary_keys.map(&:downcase).include?(col.name.downcase) # Skip primary keys
       if col.data_type.to_s == 'number'
         max = ("9" * (col.precision - col.scale)).to_i
 
@@ -350,11 +382,12 @@ File.open(ofile, 'w') do |fh|
   else
     # Character fields, size
     omg.column_info.each{ |col|
+      next if omg.primary_keys.map(&:downcase).include?(col.name.downcase) # Skip primary keys
 
-      data_type = col.data_type.to_s
+      data_type = col.data_type.to_s.downcase
 
       case data_type
-        when 'char', 'varchar', 'varchar2'
+        when 'char', 'varchar', 'varchar2', 'character varying', 'text', 'bpchar'
           validation = "validates :#{col.name.downcase}, "
           validation << ":length => {:maximum => #{col.data_size}}"
 
@@ -369,26 +402,55 @@ File.open(ofile, 'w') do |fh|
           validation << format_str
           validation << ",\n    :presence => #{!col.nullable?}" unless col.nullable?
           validation << "\n\n"
-        when 'number'
-          max = "9" * col.precision
-          max.insert(col.precision - col.scale, ".") if col.scale > 0
+        when 'number', 'integer', 'bigint', 'smallint', 'numeric', 'decimal', 'real', 'double precision'
+          # For PostgreSQL numeric types
+          if ['integer', 'bigint', 'smallint'].include?(data_type)
+            # Integer types
+            max = case data_type
+                  when 'smallint' then 32767
+                  when 'integer' then 2147483647
+                  when 'bigint' then 9223372036854775807
+                  end
+
+            validation = "\n  validates :#{col.name.downcase}"
+
+            unless col.nullable?
+              validation << ", :presence => #{!col.nullable?}"
+            end
 
-          validation = "\n  validates :#{col.name.downcase}"
+            if max
+              validation << ", :numericality => {"
+              validation << "\n    :less_than_or_equal_to => #{max}, "
+              validation << "\n    :greater_than_or_equal_to => -#{max},"
+              validation << "\n    :only_integer => true"
+              validation << "\n  }\n\n"
+            end
+          elsif data_type == 'number' || ['numeric', 'decimal', 'real', 'double precision'].include?(data_type)
+            # Oracle NUMBER or PostgreSQL decimal types
+            validation = "\n  validates :#{col.name.downcase}"
 
-          unless col.nullable?
-            validation << ", :presence => #{!col.nullable?}"
-          end
+            unless col.nullable?
+              validation << ", :presence => #{!col.nullable?}"
+            end
 
-          unless max.empty?
-            validation << ", :numericality => {"
-            validation << "\n    :less_than_or_equal_to => #{max}, "
-            validation << "\n    :greater_than_or_equal_to => -#{max}"
+            if col.respond_to?(:precision) && col.precision
+              max = "9" * col.precision
+              max.insert(col.precision - (col.scale || 0), ".") if col.scale && col.scale > 0
 
-            if col.scale == 0
-              validation << ",\n    :only_integer => true"
-            end
+              unless max.empty?
+                validation << ", :numericality => {"
+                validation << "\n    :less_than_or_equal_to => #{max}, "
+                validation << "\n    :greater_than_or_equal_to => -#{max}"
 
-            validation << "\n  }\n\n"
+                if col.scale == 0
+                  validation << ",\n    :only_integer => true"
+                end
+
+                validation << "\n  }\n\n"
+              end
+            else
+              validation << ", :numericality => true\n\n"
+            end
           end
 
       end
@@ -402,23 +464,21 @@ File.open(ofile, 'w') do |fh|
 
   # Date fields
   omg.column_info.each{ |col|
-    data_type = col.data_type.to_s
+    next if omg.primary_keys.map(&:downcase).include?(col.name.downcase) # Skip primary keys
+    data_type = col.data_type.to_s.downcase
 
-    if ['date', 'time', 'datetime'].include?(data_type)
+    if ['date', 'time', 'datetime', 'timestamp', 'timestamp without time zone', 'timestamp with time zone', 'time without time zone', 'time with time zone'].include?(data_type)
       if data_type == 'date'
         validation = "validates_date :#{col.name.downcase}"
-      end
-
-      if data_type == 'timestamp'
+      elsif data_type =~ /^time/
         validation = "validates_time :#{col.name.downcase}"
-      end
-
-      if data_type == 'datetime'
+      elsif data_type =~ /timestamp|datetime/
         validation = "validates_datetime :#{col.name.downcase}"
       end
 
       unless header_printed
         fh.puts "  # Requires the validates_timeliness library"
+        header_printed = true
       end
 
       fh.puts "  #{validation}"

data/database-model-generator.gemspec

@@ -2,7 +2,7 @@ require 'rubygems'
 
 Gem::Specification.new do |spec|
   spec.name        = 'database-model-generator'
-  spec.version     = '0.6.1'
+  spec.version     = '0.7.0'
   spec.author      = 'Daniel J. Berger'
   spec.license     = 'Apache-2.0'
   spec.email       = 'djberg96@gmail.com'
@@ -22,6 +22,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency('rspec', '~> 3.12')
   spec.add_development_dependency('ruby-oci8', '~> 2.2')
   spec.add_development_dependency('tiny_tds', '~> 3.2.1')
+  spec.add_development_dependency('pg', '~> 1.5')
 
   # Again, I only set this as a development dependency mainly as a reminder.
   # The version you need will depend on which version of Rails you're using.
@@ -29,7 +30,7 @@ Gem::Specification.new do |spec|
 
   spec.description = <<-EOF
     The database-model-generator library allows you to generate an ActiveRecord
-    model from an existing Oracle table or view, as well as automatically
-    generate a baseline test file for test-unit or minitest.
+    model from an existing Oracle, SQL Server, or PostgreSQL table or view,
+    as well as automatically generate a baseline test file for test-unit or minitest.
   EOF
 end

data/docker/postgresql/Dockerfile

@@ -0,0 +1,22 @@
+FROM ruby:3.2-alpine
+
+# Install build dependencies
+RUN apk add --no-cache \
+    build-base \
+    postgresql-dev \
+    git
+
+# Set working directory
+WORKDIR /app
+
+# Copy gemspec and Gemfile
+COPY database-model-generator.gemspec Gemfile ./
+COPY lib/database_model_generator.rb lib/
+
+# Install Ruby dependencies
+RUN bundle install
+
+# Copy the rest of the application
+COPY . .
+
+CMD ["bundle", "exec", "rspec", "spec/postgresql_model_generator_spec.rb"]