bignose
/
ledger-cli


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197
							# -*- coding: utf-8 -*-

import ledger
import cgi
import sys
import types
import posixpath
import urllib
import shutil
import os
import re

from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
from os.path import exists, join, isfile

from Cheetah.Template import Template
from Cheetah.Filters import Filter, WebSafe

webroot = join(os.getcwd(), 'python', 'res')

class UnicodeFilter(Filter):
    def filter(self, s, **kargs):
        return Filter.filter(self, s, str=unicode, **kargs)

def strip(value):
    #return re.sub('\n', '<br />', value.strip_annotations().to_string())
    return value.strip_annotations().to_string()

templateDef = '''#encoding utf-8
 <html>
    <head>
      <meta http-equiv="Content-Type" content="text/html; charset=utf-8">
      <title>$title</title>
      <link rel="stylesheet" href="/style.css" type="text/css" media="print, projection, screen" />
      <script type="text/javascript" src="/jquery-latest.js"></script>
      <script type="text/javascript" src="/jquery.tablesorter.min.js"></script>
      <script type="text/javascript" src="/jquery.tablesorter.pager.js"></script>
      <script type="text/javascript" src="/jquery.dimensions.min.js"></script>
      <script type="text/javascript">
        \$(function() { 
          \$("table")
              .tablesorter({textExtraction: 'complex',
                            widthFixed:     true,
                            widgets:        ['zebra']}) 
              .tablesorterPager({size: 100,
                                 container: \$("\#pager")});
        }); 
      </script>
    </head>
    <body>
      <div id="main">
      <h1>Register report</h1>
      <table id="register" cellspacing="1" class="tablesorter">
        <thead>
          <tr>
            <th>Date</th>
            <th>Payee</th>
            <th>Account</th>
            <th>Amount</th>
            <th>Total</th>
          </tr>
        </thead>
        <tfoot>
          <tr>
            <th>Date</th>
            <th>Payee</th>
            <th>Account</th>
            <th>Amount</th>
            <th>Total</th>
          </tr>
        </tfoot>
        <tbody>
          #for $post in $posts
          #set $total = $total + $post.amount
          <tr>
            <!--<td>${$post.xact.date if $post.xact is not $last_xact else $empty}</td>
            <td>${$post.xact.payee if $post.xact is not $last_xact else $empty}</td>-->
            <td>$post.xact.date</td>
            <td>$post.xact.payee</td>
            <td>$post.account</td>
            <td>${strip($post.amount)}</td>
            <td>${strip($total)}</td>
          </tr>
          #set $last_xact = $post.xact
          #end for
        </tbody>
      </table>
      <div id="pager" class="pager">
        <form>
          <img src="/icons/first.png" class="first"/>
          <img src="/icons/prev.png" class="prev"/>
          <input type="text" class="pagedisplay"/>
          <img src="/icons/next.png" class="next"/>
          <img src="/icons/last.png" class="last"/>
          <select class="pagesize">
            <option selected="selected" value="40">40</option>
            <option value="100">100</option>
            <option value="200">200</option>
            <option  value="300">300</option>
          </select>
        </form>
      </div>
    </body>
  </html>
'''

class LedgerHandler(BaseHTTPRequestHandler):
    def __init__(self, *args):
        self.journal = ledger.Journal(sys.argv[1])
        BaseHTTPRequestHandler.__init__(self, *args)

    def do_GET(self):
        path = self.translate_path(self.path)

        if path and exists(path) and isfile(path):
            self.copyfile(open(path), self.wfile)
        else:
            tmpl = Template(templateDef, filter=UnicodeFilter)

            tmpl.title     = 'Ledger Journal'
            tmpl.posts     = self.journal.collect(sys.argv[2])
            tmpl.total     = ledger.Value(0)
            tmpl.strip     = strip
            tmpl.last_xact = None
            tmpl.empty     = ""

            html = unicode(tmpl)
            html = html.encode('utf-8')
            self.wfile.write(html)

    def do_POST(self):
        print "Saw a POST request!"
        try:
            ctype, pdict = cgi.parse_header(self.headers.getheader('content-type'))
            if ctype == 'multipart/form-data':
                query = cgi.parse_multipart(self.rfile, pdict)
            self.send_response(301)
            self.end_headers()
        except Exception:
            print "Saw exception in POST handler"

    # This code is straight from SimpleHTTPServer.py
    def copyfile(self, source, outputfile):
        """Copy all data between two file objects.

        The SOURCE argument is a file object open for reading
        (or anything with a read() method) and the DESTINATION
        argument is a file object open for writing (or
        anything with a write() method).

        The only reason for overriding this would be to change
        the block size or perhaps to replace newlines by CRLF
        -- note however that this the default server uses this
        to copy binary data as well.

        """
        shutil.copyfileobj(source, outputfile)

    def translate_path(self, path):
        """Translate a /-separated PATH to the local filename syntax.

        Components that mean special things to the local file system
        (e.g. drive or directory names) are ignored.  (XXX They should
        probably be diagnosed.)

        """
        # abandon query parameters
        path  = path.split('?',1)[0]
        path  = path.split('#',1)[0]
        path  = posixpath.normpath(urllib.unquote(path))
        words = path.split('/')
        words = filter(None, words)
        path  = webroot
        for word in words:
            drive, word = os.path.splitdrive(word)
            head, word = os.path.split(word)
            if word in (os.curdir, os.pardir): continue
            path = os.path.join(path, word)
        return path

def main(*args):
    try:
        port   = 9000
        server = HTTPServer(('', port), LedgerHandler)
        print "Local HTTP server listening on port %d... (Control-C to exit)" \
            % port
        server.serve_forever()
    except KeyboardInterrupt:
        print "Shutting down server"
        server.socket.close()

if __name__ == '__main__':
    if len(sys.argv) < 3:
        print "usage: server.py <DATA-FILE> <REPORT-QUERY>"
        sys.exit(1)
    main()