bin/sqlt-graph

   1 #!/usr/bin/perl
   2
   3 # -------------------------------------------------------------------
   4 # $Id$
   5 # -------------------------------------------------------------------
   6 # Copyright (C) 2002-4 SQLFairy Authors
   7 #
   8 # This program is free software; you can redistribute it and/or
   9 # modify it under the terms of the GNU General Public License as
  10 # published by the Free Software Foundation; version 2.
  11 #
  12 # This program is distributed in the hope that it will be useful, but
  13 # WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15 # General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program; if not, write to the Free Software
  19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  20 # 02111-1307  USA
  21 # -------------------------------------------------------------------
  22
  23 =head1 NAME
  24
  25 sqlt-graph - Automatically create a graph from a database schema
  26
  27 =head1 SYNOPSIS
  28
  29   ./sqlt-graph -d|--db|-f|--from=db_parser [options] schema.sql
  30
  31   Options:
  32
  33     -l|--layout        Layout schema for GraphViz
  34                        ("dot," "neato," "twopi"; default "dot")
  35     -n|--node-shape    Shape of the nodes ("record," "plaintext,"
  36                        "ellipse," "circle," "egg," "triangle," "box,"
  37                        "diamond," "trapezium," "parallelogram," "house,"
  38                        "hexagon," "octagon," default "record")
  39     -o|--output        Output file name (default STDOUT)
  40     -t|--output-type   Output file type ("canon", "text," "ps," "hpgl,"
  41                        "pcl," "mif," "pic," "gd," "gd2," "gif," "jpeg,"
  42                        "png," "wbmp," "cmap," "ismap," "imap," "vrml,"
  43                        "vtx," "mp," "fig," "svg," "plain," default "png")
  44     -c|--color         Add colors
  45     --no-fields        Don't show field names
  46     --height           Image height (in inches, default "11",
  47                        set to "0" to undefine)
  48     --width            Image width (in inches, default "8.5",
  49                        set to "0" to undefine)
  50     --fontsize         custom font size for node and edge labels
  51     --fontname         name of custom font (or full path to font file) for
  52                        node, edge, and graph labels
  53     --nodeattr         attribute name and value (in key=val syntax) for
  54                        nodes; this option may be repeated to specify
  55                        multiple node attributes
  56     --edgeattr         same as --nodeattr, but for edge attributes
  57     --graphattr        same as --nodeattr, but for graph attributes
  58     --natural-join     Perform natural joins
  59     --natural-join-pk  Perform natural joins from primary keys only
  60     --show-datatypes   Show datatype of each field
  61     --show-sizes       Show column sizes for VARCHAR and CHAR fields
  62     --show-constraints Show list of constraints for each field
  63     -s|--skip          Fields to skip in natural joins
  64     --debug            Print debugging information
  65
  66 =head1 DESCRIPTION
  67
  68 This script will create a graph of your schema.  Only the database
  69 driver argument (for SQL::Translator) is required.  If no output file
  70 name is given, then image will be printed to STDOUT, so you should
  71 redirect the output into a file.
  72
  73 The default action is to assume the presence of foreign key
  74 relationships defined via "REFERNCES" or "FOREIGN KEY" constraints on
  75 the tables.  If you are parsing the schema of a file that does not
  76 have these, you will find the natural join options helpful.  With
  77 natural joins, like-named fields will be considered foreign keys.
  78 This can prove too permissive, however, as you probably don't want a
  79 field called "name" to be considered a foreign key, so you could
  80 include it in the "skip" option, and all fields called "name" will be
  81 excluded from natural joins.  A more efficient method, however, might
  82 be to simply deduce the foriegn keys from primary keys to other fields
  83 named the same in other tables.  Use the "natural-join-pk" option
  84 to acheive this.
  85
  86 If the schema defines foreign keys, then the graph produced will be
  87 directed showing the direction of the relationship.  If the foreign
  88 keys are intuited via natural joins, the graph will be undirected.
  89
  90 =cut
  91
  92 # -------------------------------------------------------------------
  93
  94 use strict;
  95 use Data::Dumper;
  96 use Getopt::Long;
  97 use GraphViz;
  98 use Pod::Usage;
  99 use SQL::Translator;
 100
 101 use vars '$VERSION';
 102 $VERSION = sprintf "%d.%02d", q$Revision$ =~ /(\d+)\.(\d+)/;
 103
 104 #
 105 # Get arguments.
 106 #
 107 my (
 108     $layout, $node_shape, $out_file, $output_type, $db_driver, $add_color,
 109     $natural_join, $join_pk_only, $skip_fields, $show_datatypes,
 110     $show_sizes, $show_constraints, $debug, $help, $height, $width,
 111     $no_fields, $fontsize, $fontname
 112 );
 113
 114 # multi-valued options:
 115 my %edgeattrs = ();
 116 my %nodeattrs = ();
 117 my %graphattrs = ();
 118
 119 GetOptions(
 120     'd|db|f|from=s'    => \$db_driver,
 121     'o|output:s'       => \$out_file,
 122     'l|layout:s'       => \$layout,
 123     'n|node-shape:s'   => \$node_shape,
 124     't|output-type:s'  => \$output_type,
 125     'height:f'         => \$height,
 126     'width:f'          => \$width,
 127     'fontsize=i'       => \$fontsize,
 128     'fontname=s'       => \$fontname,
 129     'nodeattr=s'       => \%nodeattrs,
 130     'edgeattr=s'       => \%edgeattrs,
 131     'graphattr=s'      => \%graphattrs,
 132     'c|color'          => \$add_color,
 133     'no-fields'        => \$no_fields,
 134     'natural-join'     => \$natural_join,
 135     'natural-join-pk'  => \$join_pk_only,
 136     's|skip:s'         => \$skip_fields,
 137     'show-datatypes'   => \$show_datatypes,
 138     'show-sizes'       => \$show_sizes,
 139     'show-constraints' => \$show_constraints,
 140     'debug'            => \$debug,
 141     'h|help'           => \$help,
 142 ) or die pod2usage;
 143 my @files = @ARGV; # the create script(s) for the original db
 144
 145 pod2usage(1) if $help;
 146 pod2usage( -message => "No db driver specified" ) unless $db_driver;
 147 pod2usage( -message => 'No input file'          ) unless @files;
 148
 149 my $translator           =  SQL::Translator->new(
 150     from                 => $db_driver,
 151     to                   => 'GraphViz',
 152     debug                => $debug || 0,
 153     producer_args        => {
 154         out_file         => $out_file,
 155         layout           => $layout,
 156         node_shape       => $node_shape,
 157         output_type      => $output_type,
 158         add_color        => $add_color,
 159         natural_join     => $natural_join,
 160         natural_join_pk  => $join_pk_only,
 161         skip_fields      => $skip_fields,
 162         show_datatypes   => $show_datatypes,
 163         show_sizes       => $show_sizes,
 164         show_constraints => $show_constraints,
 165         height           => $height || 0,
 166         width            => $width  || 0,
 167         fontsize         => $fontsize,
 168         fontname         => $fontname,
 169         nodeattrs        => \%nodeattrs,
 170         edgeattrs        => \%edgeattrs,
 171         graphattrs       => \%graphattrs,
 172         show_fields      => $no_fields ? 0 : 1,
 173     },
 174 ) or die SQL::Translator->error;
 175
 176 for my $file (@files) {
 177     my $output = $translator->translate( $file ) or die
 178                  "Error: " . $translator->error;
 179     if ( $out_file ) {
 180         print "Image written to '$out_file'.  Done.\n";
 181     }
 182     else {
 183         print $output;
 184     }
 185 }
 186
 187 # -------------------------------------------------------------------
 188
 189 =pod
 190
 191 =head1 AUTHOR
 192
 193 Ken Y. Clark E<lt>kclark@cpan.orgE<gt>.
 194
 195 =head1 SEE ALSO
 196
 197 perl, SQL::Translator.
 198
 199 =cut