diff options
| -rwxr-xr-x | pg2rb.rb | 153 |
1 files changed, 153 insertions, 0 deletions
diff --git a/pg2rb.rb b/pg2rb.rb new file mode 100755 index 0000000..764c94e --- /dev/null +++ b/pg2rb.rb @@ -0,0 +1,153 @@ +#! /usr/bin/env ruby +# -*- coding: UTF-8 -*- + +#---------------------------------------------------------------------------- +# +# File : dia2pg.rb +# Author : Jérémy Zurcher <jeremy@asynk.ch> +# Date : 24/08/10 +# License : +# +# Permission is hereby granted, free of charge, to any person obtaining +# a copy of this software and associated documentation files (the +# "Software"), to deal in the Software without restriction, including +# without limitation the rights to use, copy, modify, merge, publish, +# distribute, sublicense, and/or sell copies of the Software, and to +# permit persons to whom the Software is furnished to do so, subject to +# the following conditions: +# +# The above copyright notice and this permission notice shall be +# included in all copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, +# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF +# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND +# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE +# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION +# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION +# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. +# +#---------------------------------------------------------------------------- +# +require 'ostruct' +require 'optparse' +# +class OpenStruct + def to_h; @table end +end +# +options = OpenStruct.new +# +options.verbose = false +options.input = nil +options.database = nil +options.indent=4 +# +opts = OptionParser.new do |opts| + opts.banner = "Usage: #{$0} [options]" + opts.on( "-i", "--input FILENAME", "input file" ) { |n| options.input = n } + opts.on( "-V", "--verbose", "produce verbose output" ) { options.verbose = true } + opts.on( "-d", "--database DATABASE", "set default database" ) { |n| options.database = n } + opts.on( "-n", "--indent N", "set number of indentation spaces" ) { |n| options.indent = n.to_i } +end +# +options.tbl_prefix+='_' if not options.tbl_prefix.nil? and options.tbl_prefix[-1]!='_' +# +opts.parse!(ARGV) +# +if options.input.nil? + $stderr << "missing input file argument\n" + exit 1 +end +# +INDENT="\n"+" "*options.indent +# +class Table + # + def initialize n, t, a, d + @name = n + @tbl_prefix = t + @attr_prefix = a + @db = d + @ids = [] + @bools = [] + @floats = [] + @integers = [] + @texts = [] + @times = [] + end + attr_reader :ids, :bools, :floats, :integers, :texts, :times + # + def name + (@tbl_prefix.nil? ? @name : @name[@tbl_prefix.length..-1]).capitalize # TODO replace _(.) by $1.capitalize + end + # + def to_s + r="class #{name}" + r << INDENT+'#' + r << INDENT+'include Hmsa_lib::PgsqlModel' + r << INDENT+'#' + r << INDENT+"db '#{@db}'.freeze" unless @db.nil? + r << INDENT+"prefix '#{@attr_prefix}'.freeze" unless @attr_prefix.nil? + r << INDENT+'#' unless @db.nil? and @attr_prefix.nil? + r << INDENT+"ids " + @ids.collect { |i| ":#{i}" }.join(', ') unless @ids.empty? + [ :bools, :floats, :integers, :texts, :times].each do |sym| + attrs = send sym + next if attrs.empty? + r << INDENT+"#{sym} " + attrs.collect { |a| ":#{@attr_prefix.nil? ? a : a[@attr_prefix.length..-1]}" }.join(', ') + end + r << INDENT+'#' + r << INDENT+'sql_stmts( {' + r << INDENT+" # :exists => [ 'select count(*) from #{@name} where #{@ids[0]}=$1::int', :#{@ids[0]} ]," unless @ids.empty? + r << INDENT+'} )' + r << INDENT+'#' + r << "\nend" + end +end +# +TABLES=[] +# +tbl=nil +tbl_prefix=nil +attr_prefix=nil +File.open(options.input).readlines.each do |l| +# puts l + if l=~/dia2pg/ + l.split.each do |s| + k,v = s.split '=' + if k=~/tbl_prefix/ + tbl_prefix=v + elsif k=~/attr_prefix/ + attr_prefix=v + end + end + elsif l=~/CREATE TABLE (\S+)/ + tbl = Table.new $1, tbl_prefix, attr_prefix, options.database + tbl_prefix=nil + attr_prefix=nil + TABLES << tbl + elsif l=~/PRIMARY\s+KEY\s+\((\w*)\)/ + $1.split(/,/).reverse.each do |i| + tbl.ids.insert 0, i + end + elsif l=~/(\w+)\s+integer\s+REFERENCES/ + tbl.ids << $1 + elsif l=~/(\w+)\s+boolean/ + tbl.bools << $1 + elsif l=~/(\w+)\s+double/ + tbl.floats << $1 + elsif l=~/(\w+)\s+integer/ + tbl.integers << $1 + elsif l=~/(\w+)\s+text/ + tbl.texts << $1 + elsif l=~/(\w+)\s+char/ + tbl.texts << $1 + elsif l=~/(\w+)\s+timestamp/ + tbl.times << $1 + end +end +# +TABLES.each do |t| + puts "#" + puts t +end |