コード例 #1
0
ファイル: pdfwriter.py プロジェクト: jbq/pancito
    def gencontract(self, user, contract):
        c = self.conn.cursor()

        c.execute("SELECT coalesce(sum(amount), 0) FROM extra_payment WHERE contract_id = ? AND user_id = ?", (contract['id'], user['id']))
        extraAmount = c.fetchone()[0]

        c.execute("SELECT sum(quantity) AS quantity, * FROM bakeorder INNER JOIN bake ON bake.rowid = bakeid INNER JOIN product ON product.id = productid WHERE contract_id = ? AND userid = ? GROUP BY productid", (contract['id'], user['id']))
        orders = c.fetchall()
        if len(orders) == 0:
            raise Exception("No order for %s, skipping" % user['name'])

        # Adapt contract for display
        for field in ('startdate', 'enddate'):
            contract[field] = contract[field].strftime("%d %B %Y")

        info = Struct(**user)
        info.update(**contract)
        info.placeName = contract['place']['name']
        info.date = date.today().strftime("%d %B %Y")
        c.execute("SELECT count(*) FROM bake WHERE contract_id = ?", (contract['id'],))
        info.bakeCount = c.fetchone()[0]

        # FIXME
        orderdisplays = []
        orderAmount = 0
        for order in orders:
            orderdisplays.append("%s %s" % (order['quantity'], order['name']))
            orderAmount += order['quantity'] * order['itemprice']
        info.order = " et ".join(orderdisplays)
        info.balance = pancito.displayAmount(orderAmount - extraAmount)
        syslog.syslog(syslog.LOG_DEBUG, "Contract info: %s" % repr(info.__dict__))

        rw_database = RWDatabase(fs=lfs)
        handler = rw_database.get_handler(os.path.join(pancito.datadir, 'model.odt'))
        document = stl_to_odt(handler, info)
        handler = ODTFile(string=document)
        tmpfile = tempfile.mktemp(".odt")
        tmpdir = tempfile.mkdtemp("", "pdfwriter")
        rw_database.set_handler(tmpfile, handler)
        rw_database.save_changes()

        if not os.path.exists("/usr/bin/libreoffice"):
            raise Exception("Cannot find libreoffice")

        if not os.path.exists("/usr/share/fonts/truetype/msttcorefonts/Arial.ttf"):
            raise Exception("Arial font not found, please install ttf-mscorefonts-installer")

        cmd = ["/usr/bin/libreoffice", "--headless", "--invisible", "--convert-to", "pdf", "--outdir", tmpdir, tmpfile]
        p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        sc = p.wait()
        if sc != 0:
            raise Exception("Command returned status %s: %s.\nOutput:\n%s\nError:\n%s" % (sc, cmd, p.stdout.read(), p.stderr.read()))

        os.unlink(tmpfile)
        self.createAdhesion(user['id'], contract['id'], orderAmount)
        return os.path.join(tmpdir, os.path.basename(tmpfile).replace('.odt', '.pdf'))