Properly filter year
[infodrom.org/service.infodrom.org] / src / InfoCon / buch / infocon
index 5f1f86a..a8bfaca 100755 (executable)
@@ -1,7 +1,7 @@
 #! /usr/bin/perl
 
-#  infocon - Admin-Tool for InfoCon
-#  Copyright (c) 1998-2003,2005-6  Martin Schulze <joey@infodrom.org>
+#  infocon - Administration tool for InfoCon
+#  Copyright (c) 1998-2003,2005-8,10  Martin Schulze <joey@infodrom.org>
 #
 #  This program is free software; you can redistribute it and/or modify
 #  it under the terms of the GNU General Public License as published by
 
 # $Id$
 
+use strict;
+use warnings;
+
 use DBI;
 use Term::ReadLine;
+use Getopt::Long;
 
-$table = "sales";
-$engine  = "dbi:Pg:dbname=infocon";
-$dbh = DBI->connect($engine);
+my $table = "sales";
+my $engine  = "dbi:Pg:dbname=infocon";
+my $dbh = DBI->connect($engine);
 if (!$dbh) {
     print "Access to database denied!\n";
     return 1;
 }
 
-@categories = ();
+my %data;
+my @categories = ();
+my $term = undef;
+my $opt_all = 0;
+my $opt_verbose = 0;
+my $opt_year = 0;
+my $opt_direction = undef;
 
 sub sdate
 {
@@ -43,10 +53,12 @@ sub sdate
 #
 sub date_to_string
 {
+    my ($day,$mon,$year);
+
     return "" if (!$_[0]);
 
-    ($date_sec,$date_min,$date_hour,$date_mday,$date_mon,$date_year,$date_wday,$date_isdst)
-       = localtime();
+    my ($date_sec,$date_min,$date_hour,$date_mday,$date_mon,$date_year,$date_wday,$date_isdst)
+       = localtime;
 
     if ($_[0] eq "heute" || $_[0] eq "sofort" || $_[0] eq "pronto" || $_[0] eq "today" || $_[0] eq "now") {
        $day = $date_mday;
@@ -110,7 +122,7 @@ sub sales_list
     my $sth;
     my $d;
 
-    if ($where !~ /visible/ && (!$opt_all || $opt_all == 0)) {
+    if ($where && $where !~ /visible/ && (!$opt_all || $opt_all == 0)) {
        if ($where) {
            $where .= " AND visible = 1";
        } else {
@@ -120,9 +132,9 @@ sub sales_list
 
     if ($opt_year) {
        if ($where) {
-           $where .= " AND date ~* '$opt_year'";
+           $where .= " AND date LIKE '$opt_year%'";
        } else {
-           $where .= "date ~* '$opt_year'";
+           $where .= "date LIKE '$opt_year%'";
        }
     }
 
@@ -140,16 +152,22 @@ sub sales_list
        }
     }
 
+    if ($where !~ /visible/) {
+       $where .= " AND " if $where;
+       $where .= "visible = 1";
+    }
+
     $query  = "SELECT nr,date,description,price FROM $table";
     $query .= " WHERE $where" if ($where);
     $query .= " ORDER by date,nr";
+
     $sth = $dbh->prepare($query);
-    if ($sth && ($rc = $sth->execute) > 0) {
+    if ($sth && (my $rc = $sth->execute) > 0) {
        print " Nr.   Datum  Bezeichnung                                 Betrag\n";
        print "------------------------------------------------------------------\n";
        while (@row = $sth->fetchrow_array) {
            $descr = substr($row[2],0,40);
-           printf "%4d %8s %-40s  %9.2f\n", $row[0], &sdate($row[1]), $descr, $row[3];
+           printf "%4d %8s %-40s  %9.2f\n", $row[0], sdate($row[1]), $descr, $row[3];
            if ($row[3] < 0.0) {
                $sum_neg -= $row[3];
            } else {
@@ -165,6 +183,30 @@ sub sales_list
        print "==================================================================\n";
        printf " Summe                                                  %9.2f\n\n", $sum_pos - $sum_neg;
     }
+    $data{'done'} = 1;
+}
+
+sub get_descriptions
+{
+    my $category = shift;
+    my $query;
+    my @row;
+    my $sth;
+    my @arr = ();
+    my ($date_sec,$date_min,$date_hour,$date_mday,$date_mon,$date_year,$date_wday,$date_isdst)
+       = localtime;
+
+    $query  = sprintf("SELECT DISTINCT description FROM %s WHERE date LIKE '%d%%' ",
+                     $table, $date_year+1900);
+    $query .= sprintf("AND category = '%s' ", $category) if defined $category;
+    $query .= "ORDER by description";
+    $sth = $dbh->prepare($query);
+    if ($sth && (my $rc = $sth->execute) > 0) {
+       while (@row = $sth->fetchrow_array) {
+           push(@arr, $row[0]) if ($row[0]);
+       }
+    }
+    return @arr;
 }
 
 sub get_categories
@@ -176,7 +218,7 @@ sub get_categories
 
     $query  = "SELECT DISTINCT category FROM $table ORDER by category";
     $sth = $dbh->prepare($query);
-    if ($sth && ($rc = $sth->execute) > 0) {
+    if ($sth && (my $rc = $sth->execute) > 0) {
        while (@row = $sth->fetchrow_array) {
            push(@arr, $row[0]) if ($row[0]);
        }
@@ -186,9 +228,11 @@ sub get_categories
 
 sub list_categories
 {
-    @categories = &get_categories() if ($#categories);
+    @categories = get_categories unless @categories;
 
     printf "%s\n", join (", ",@categories);
+
+    exit;
 }
 
 sub read_input
@@ -207,6 +251,7 @@ sub read_input
     } elsif ($ans eq ".") {
        $ans = '';
     }
+    $ans =~ s/ *$// if $ans;
     return $ans;
 }
 
@@ -238,16 +283,44 @@ sub buchung_input
     my @input = ();
     my $weiter = 'y';
     my $i;
+    my $ans;
     my $query;
+    my @descriptions;
     my ($date_sec,$date_min,$date_hour,$date_mday,$date_mon,$date_year,$date_wday,$date_isdst)
-       = localtime();
+       = localtime;
+
+    @categories = get_categories unless @categories;
+    @descriptions = get_descriptions;
+
+    $term = new Term::ReadLine '' unless $term;
+
+    $term->addhistory($_) foreach (@categories);
+    $term->addhistory($_) foreach (@descriptions);
+
+    my $attribs = $term->Attribs;
+
+    my $sth = $dbh->prepare ("INSERT INTO $table VALUES (?,?,?,?,?,?,?,?,?)");
 
     print "Buchungseingabe\n\n";
     while ($weiter =~ /[JjYy1]/) {
        $i=0;while ($i <= $#fieldname) {
-           $ans = &read_input($fieldname[$i],$input[$i]);
+           if ($fieldname[$i] eq "Category") {
+               $attribs->{completion_entry_function} = $attribs->{list_completion_function};
+               $attribs->{completion_word} = \@categories;
+           } elsif ($fieldname[$i] eq "Description") {
+               $attribs->{completion_entry_function} = $attribs->{list_completion_function};
+               @descriptions = get_descriptions $input[1];
+               $attribs->{completion_word} = \@descriptions;
+           } elsif ($fieldname[$i] eq "Tax assigned") {
+               if ($input[$i-1] == 0) {
+                   $input[$i++] = 0;
+                   next;
+               }
+           } else {
+               $attribs->{completion_word} = undef;
+           }
+           $ans = read_input($fieldname[$i],$input[$i]);
            if ($fieldname[$i] eq "Category" && $ans eq "?") {
-               @categories = &get_categories() if ($#categories);
                printf "  %s\n", join (", ",@categories);
            } elsif ($fieldname[$i] eq "Datum") {
                if ($ans =~ /^\d+\.\d+.\d+$/) {
@@ -268,6 +341,7 @@ sub buchung_input
                } else {
                    $input[$i] = 0;
                }
+               $i++;
            } else {
                $input[$i] = $ans;
                $i++;
@@ -287,13 +361,31 @@ sub buchung_input
            $input[6] *= -1 if ($input[6] > 0);
        }
 
-       $query = sprintf ("INSERT INTO $table VALUES (%d,'%s','%s','%s',%8.2f,%8.2f,%8.2f,%d,%d)",
-                         &get_next_nr(), &date_to_string($input[0]), $input[1], $input[2], $input[4],
+       $sth->execute (get_next_nr(), date_to_string($input[0]), $input[1], $input[2], $input[4],
                          $input[5], $input[6], 1, $input[7]);
-       $sth = $dbh->do($query);
-       $weiter = &read_input("Weiter",'j');
+       $weiter = read_input("Weiter",'j');
        $input[5] = 0.0;
     }
+
+    exit;
+}
+
+sub buchung_hidden
+{
+    $table = "sales_dm";
+    sales_list("visible = 0");
+    $table = "sales";
+    sales_list("visible = 0");
+    exit;
+}
+
+sub buchung_unpaid
+{
+    $table = "sales_dm";
+    sales_list("paid = 0");
+    $table = "sales";
+    sales_list("paid = 0");
+    exit;
 }
 
 sub usage
@@ -315,114 +407,85 @@ sub usage
     exit 0;
 }
 
-$term = new Term::ReadLine '';
-$i = 0;
-$opt_all = 0;
-$opt_verbose = 0;
-$opt_year = 0;
-&usage() if ($#ARGV == -1);
-while ($i <= $#ARGV) {
-    # Some aliases
-    if ($ARGV[$i] eq "-bc") {
-       $ARGV[$i] = "--buchung-category";
-    } elsif ($ARGV[$i] eq "-bi") {
-       $ARGV[$i] = "--buchung-input";
-    } elsif ($ARGV[$i] eq "-bh") {
-       $ARGV[$i] = "--buchung-hidden";
-    } elsif ($ARGV[$i] eq "-bu") {
-       $ARGV[$i] = "--buchung-unpaid";
-    } elsif ($ARGV[$i] eq "-lc") {
-       $ARGV[$i] = "--list-categories";
+%data = (
+    'category' => undef,
+    'done' => undef,
+    'pay' => undef,
+    'unpay' => undef,
+    'hide' => undef,
+    'unhide' => undef,
+    'mailto' => undef,
+    'buchung-input' => undef,
+    'buchung-unpaid' => undef,
+    'buchung-hidden' => undef,
+    'list-categories' => undef,
+    );
+my %options = (
+    'buchung-category|bc:s' => \$data{category},
+    'pay=s' => \$data{pay},
+    'unpay=s' => \$data{unpay},
+    'hide=s' => \$data{hide},
+    'unhide=s' => \$data{unhide},
+    'year=i' => \$opt_year,
+    'direction|d=s' => \$opt_direction,
+    'mailto:s' => \$data{mailto},
+    'all' => \$opt_all,
+    'verbose' => \$opt_verbose,
+    'help' => \&usage,
+    'dm' => sub {$table = "sales_dm"},
+    'buchung-input|bi' => \$data{'buchung-input'},
+    'buchung-unpaid|bu' => \$data{'buchung-unpaid'},
+    'buchung-hidden|bh' => \$data{'buchung-hidden'},
+    'list-categories|lc' => \$data{'list-categories'},
+    );
+
+my $cmdln = 'infocon ' . join (' ', @ARGV);
+GetOptions(%options);
+
+if ($opt_year != 0 && $opt_year < 2002) {
+    $table = "sales_dm";
+}
+
+if (defined $opt_direction) {
+    usage unless $opt_direction =~ /^(in|out)$/i;
+}
+
+if (defined $data{mailto}) {
+    if (open(STDOUT, "| /usr/sbin/sendmail -t")) {
+       print  "From: Joey Schulze <joey\@infodrom.org>\n";
+       printf "To: %s\n", length($data{mailto})?$data{mailto}:'Joey Schulze <joey@infodrom.org>';
+       printf "Subject: %s\n", $cmdln;
+       print  "MIME-Version: 1.0\n";
+       print  "Content-type: text/plain; charset=iso-8859-1\n";
+       print  "Content-Disposition: inline\n";
+       print  "Content-Transfer-Encoding: 8bit\n";
+       print  "\n";
     }
+}
 
-    if ($ARGV[$i] eq "-h" || $ARGV[$i] eq "--help") {
-       &usage();
-    } elsif ($ARGV[$i] =~ /^--list-/) {
-       $ARGV[$i] =~ s/^--list-//;
-       if ($ARGV[$i] eq "categories") {
-           &list_categories();
-       } else {
-           &usage();
-       }
-    } elsif ($ARGV[$i] =~ /^--buchung-/) {
-       $ARGV[$i] =~ s/^--buchung-//;
-       if ($ARGV[$i] eq "category") {
-           if ($i+1 <= $#ARGV && ($ARGV[$i+1] !~ /^-/)) {
-               $i++;
-               &sales_list("category = '$ARGV[$i]'");
-           } else {
-               &sales_list();
-           }
-       } elsif ($ARGV[$i] eq "hidden") {
-           $saved_table = $table;
-           $table = "sales_dm";
-           &sales_list("visible = 0");
-           $table = "sales";
-           &sales_list("visible = 0");
-           $table = $table_saved;
-       } elsif ($ARGV[$i] eq "input") {
-           &buchung_input();
-       } elsif ($ARGV[$i] eq "unpaid") {
-           $saved_table = $table;
-           $table = "sales_dm";
-           &sales_list("paid = 0");
-           $table = "sales";
-           &sales_list("paid = 0");
-           $table = $table_saved;
-       } else {
-           &usage();
-       }
-    } elsif ($ARGV[$i] eq "--pay") {
-       if ($i+1 <= $#ARGV && ($ARGV[$i+1] !~ /^-/)
-           && ($ARGV[$i+1] =~ /^\d+$/)) {
-           $i++;
-           &pay_invoice ($ARGV[$i], 1);
-       }
-    } elsif ($ARGV[$i] eq "--unpay") {
-       if ($i+1 <= $#ARGV && ($ARGV[$i+1] !~ /^-/)
-           && ($ARGV[$i+1] =~ /^\d+$/)) {
-           $i++;
-           &pay_invoice ($ARGV[$i], 0);
-       }
-    } elsif ($ARGV[$i] eq "--hide") {
-       if ($i+1 <= $#ARGV && ($ARGV[$i+1] !~ /^-/)
-           && ($ARGV[$i+1] =~ /^\d+$/)) {
-           $i++;
-           &hide_invoice ($ARGV[$i], 1);
-       }
-    } elsif ($ARGV[$i] eq "--unhide") {
-       if ($i+1 <= $#ARGV && ($ARGV[$i+1] !~ /^-/)
-           && ($ARGV[$i+1] =~ /^\d+$/)) {
-           $i++;
-           &hide_invoice ($ARGV[$i], 0);
-       }
-    } elsif ($ARGV[$i] eq "--year" || $ARGV[$i] eq "-y") {
-       if ($i+1 <= $#ARGV && ($ARGV[$i+1] =~ /^(\d+)$/)) {
-           $i++;
-           $opt_year = $1;
-           if ($opt_year < 70) {
-               $opt_year += 2000;
-           } elsif ($opt_year < 100) {
-               $opt_year += 1900;
-           }
-           if ($opt_year < 2002) {
-               $table = "sales_dm";
-           } elsif ($opt_year > 2001) {
-               $table = "sales";
-           }
-       }
-    } elsif ($ARGV[$i] eq "--direction" || $ARGV[$i] eq "--dir"
-            || $ARGV[$i] eq "-d") {
-       if ($i+1 <= $#ARGV && ($ARGV[$i+1] =~ /^(in|out)$/i)) {
-           $i++;
-           $opt_direction = $1;
-       }
-    } elsif ($ARGV[$i] eq "--dm") {
-       $table = "sales_dm";
-    } elsif ($ARGV[$i] eq "-a" || $ARGV[$i] eq "--all") {
-       $opt_all = 1;
-    } elsif ($ARGV[$i] eq "-v" || $ARGV[$i] eq "--verbose") {
-       $opt_verbose = 1;
+if (defined $data{category}) {
+    if (length($data{category})) {
+       sales_list("category = '".$data{category}."'");
+    } else {
+       sales_list;
     }
-    $i++;
+    exit;
+} elsif (defined $data{'buchung-input'}) {
+    buchung_input;
+} elsif (defined $data{'buchung-unpaid'}) {
+    buchung_unpaid;
+} elsif (defined $data{'buchung-hidden'}) {
+    buchung_hidden;
+} elsif (defined $data{'list-categories'}) {
+    list_categories;
+} elsif (defined $data{pay}) {
+    pay_invoice($data{pay}, 1);
+} elsif (defined $data{unpay}) {
+    pay_invoice($data{unpay}, 0);
+} elsif (defined $data{hide}) {
+    hide_invoice($data{hide}, 1);
+} elsif (defined $data{unhide}) {
+    hide_invoice($data{unhide}, 0);
+} else {
+    usage;
 }