X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FSQL%2FTranslator%2FProducer%2FClassDBI.pm;h=4a93011641503c1d6c135d2a0f9157f4126d103e;hb=d4f84dd192edc7a64a0b1a9923f1bafc0bc5ef9d;hp=726eb81614bd89a2f2f845bce57a23a65e2cb8d3;hpb=420f63c737b1a6cf568d4b90e9dad1e52b8b1f2f;p=dbsrgits%2FSQL-Translator.git diff --git a/lib/SQL/Translator/Producer/ClassDBI.pm b/lib/SQL/Translator/Producer/ClassDBI.pm index 726eb81..4a93011 100755 --- a/lib/SQL/Translator/Producer/ClassDBI.pm +++ b/lib/SQL/Translator/Producer/ClassDBI.pm @@ -1,10 +1,9 @@ package SQL::Translator::Producer::ClassDBI; # ------------------------------------------------------------------- -# $Id: ClassDBI.pm,v 1.38 2003-10-15 19:56:13 allenday Exp $ +# $Id: ClassDBI.pm 1440 2009-01-17 16:31:57Z jawnsy $ # ------------------------------------------------------------------- -# Copyright (C) 2003 Allen Day , -# Ying Zhang +# Copyright (C) 2002-2009 SQLFairy Authors # # This program is free software; you can redistribute it and/or # modify it under the terms of the GNU General Public License as @@ -22,12 +21,11 @@ package SQL::Translator::Producer::ClassDBI; # ------------------------------------------------------------------- use strict; -use vars qw[ $VERSION $DEBUG ]; -$VERSION = sprintf "%d.%02d", q$Revision: 1.38 $ =~ /(\d+)\.(\d+)/; +use vars qw[ $DEBUG ]; $DEBUG = 1 unless defined $DEBUG; use SQL::Translator::Schema::Constants; -use SQL::Translator::Utils qw(header_comment); +use SQL::Translator::Utils qw(debug header_comment); use Data::Dumper; my %CDBI_auto_pkgs = ( @@ -44,9 +42,16 @@ sub produce { my $no_comments = $t->no_comments; my $schema = $t->schema; my $args = $t->producer_args; + if ( my $fmt = $args->{'format_pkg_name'} ) { + $t->format_package_name( $fmt ); + } + if ( my $fmt = $args->{'format_fk_name'} ) { + $t->format_fk_name( $fmt ); + } my $db_user = $args->{'db_user'} || ''; - my $db_pass = $args->{'db_pass'} || ''; - my $main_pkg_name = $args->{'main_pkg_name'} || + my $db_pass = $args->{'db_password'} || ''; + my $main_pkg_name = $args->{'package_name'} || + # $args->{'main_pkg_name'} || # keep this? undocumented $t->format_package_name('DBI'); my $header = header_comment( __PACKAGE__, "# " ); my $parser_type = ( split /::/, $t->parser_type )[-1]; @@ -57,14 +62,16 @@ sub produce { ); my $sep = '# ' . '-' x 67; + # # Identify "link tables" (have only PK and FK fields). # my %linkable; my %linktable; - foreach my $table ( $schema->get_tables ) { + for my $table ( $schema->get_tables ) { + debug("PKG: Table = ", $table->name, "\n"); my $is_link = 1; - foreach my $field ( $table->get_fields ) { + for my $field ( $table->get_fields ) { unless ( $field->is_primary_key or $field->is_foreign_key ) { $is_link = 0; last; @@ -106,8 +113,8 @@ sub produce { for my $table ( $schema->get_tables ) { my $table_name = $table->name or next; - my $table_pkg_name = $t->format_package_name($table_name); - $packages{$table_pkg_name} = { + my $table_pkg_name = join '::', $main_pkg_name, $t->format_package_name($table_name); + $packages{ $table_pkg_name } = { order => ++$order, pkg_name => $table_pkg_name, base => $main_pkg_name, @@ -117,18 +124,18 @@ sub produce { # # Primary key may have a differenct accessor method name # - if ( my $constraint = $table->primary_key ) { - my $field = ( $constraint->fields )[0]; - $packages{ $table_pkg_name }{'_columns_primary'} = $field; - - if ( my $pk_xform = $t->format_pk_name ) { - my $pk_name = $pk_xform->( $table_pkg_name, $field ); - - $packages{$table_pkg_name}{'pk_accessor'} = - "#\n# Primary key accessor\n#\n" - . "sub $pk_name {\n shift->$field\n}\n\n"; - } - } +# if ( my $constraint = $table->primary_key ) { +# my $field = ( $constraint->fields )[0]; +# $packages{ $table_pkg_name }{'_columns_primary'} = $field; +# +# if ( my $pk_xform = $t->format_pk_name ) { +# my $pk_name = $pk_xform->( $table_pkg_name, $field ); +# +# $packages{$table_pkg_name}{'pk_accessor'} = +# "#\n# Primary key accessor\n#\n" +# . "sub $pk_name {\n shift->$field\n}\n\n"; +# } +# } my $is_data = 0; foreach my $field ( $table->get_fields ) { @@ -263,33 +270,40 @@ sub produce { if ( $field->is_foreign_key ) { my $table_name = $table->name; my $field_name = $field->name; - my $fk_method = $t->format_fk_name( $table_name, $field_name ); +# my $fk_method = $t->format_fk_name( $table_name, $field_name ); + my $fk_method = join('::', $table_pkg_name, + $t->format_fk_name( $table_name, $field_name ) + ); my $fk = $field->foreign_key_reference; my $ref_table = $fk->reference_table; my $ref_pkg = $t->format_package_name($ref_table); my $ref_field = ( $fk->reference_fields )[0]; +# my $fk_method = join('::', +# $table_pkg_name, $t->format_fk_name( $ref_table ) +# ); push @{ $packages{$table_pkg_name}{'has_a'} }, "$table_pkg_name->has_a(\n" . " $field_name => '$ref_pkg'\n);\n\n" . "sub $fk_method {\n" - . " return shift->$field_name\n}\n\n"; + . " return shift->$field_name\n}\n\n" + ; # if there weren't M-M relationships via the has_many # being set up here, create nice pluralized method alias # rather for user as alt. to ugly tablename_fieldname name # - if ( !$packages{$ref_pkg}{'has_many'}{$table_name} ) { - # - # ADD CALLBACK FOR PLURALIZATION MANGLING HERE - # - push @{ $packages{$ref_pkg}{'has_many'}{$table_name} }, - "sub ${table_name}s {\n " . - "return shift->$table_name\_$field_name\n}\n\n"; - # else ugly - } - else { - } +# if ( !$packages{$ref_pkg}{'has_many'}{$table_name} ) { +# # +# # ADD CALLBACK FOR PLURALIZATION MANGLING HERE +# # +# push @{ $packages{$ref_pkg}{'has_many'}{$table_name} }, +# "sub ${table_name}s {\n " . +# "return shift->$table_name\_$field_name\n}\n\n"; +# # else ugly +# } +# else { +# } push @{ $packages{$ref_pkg}{'has_many'}{$table_name} }, "$ref_pkg->has_many(\n '${table_name}_${field_name}', " @@ -315,7 +329,8 @@ sub produce { sort { $packages{ $a }{'order'} <=> $packages{ $b }{'order'} } keys %packages ) { - my $pkg = $packages{$pkg_name}; + my $pkg = $packages{$pkg_name} or next; + next unless $pkg->{'pkg_name'}; $create .= join ( "\n", $sep,