add application comparoob

weboob/applications/comparoob/__init__.py

@@ -0,0 +1,7 @@
+# -*- coding: utf-8 -*-
+# vim: ft=python et softtabstop=4 cinoptions=4 shiftwidth=4 ts=4 ai
+
+
+from .comparoob import Comparoob
+
+__all__ = ['Comparoob']

weboob/applications/comparoob/comparoob.py

@@ -0,0 +1,135 @@
+# -*- coding: utf-8 -*-
+
+# Copyright(C) 2012 Romain Bignon
+#
+# This file is part of weboob.
+#
+# weboob is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# weboob is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with weboob. If not, see <http://www.gnu.org/licenses/>.
+
+
+from __future__ import with_statement
+
+import sys
+
+from weboob.capabilities.pricecomparison import ICapPriceComparison
+from weboob.tools.misc import html2text
+from weboob.tools.application.repl import ReplApplication
+from weboob.tools.application.formatters.iformatter import IFormatter
+
+
+__all__ = ['Comparoob']
+
+
+class PriceFormatter(IFormatter):
+    MANDATORY_FIELDS = ('id', 'cost', 'currency', 'shop', 'product')
+
+    def flush(self):
+        pass
+
+    def format_dict(self, item):
+        if item['message']:
+            message = item['message']
+        else:
+            message = '%s (%s)' % (item['shop'].name, item['shop'].location)
+
+        result = u'%s%s%s\n' % (self.BOLD, message, self.NC)
+        result += 'ID: %s\n' % item['id']
+        result += 'Product: %s\n' % item['product'].name
+        result += 'Cost: %s%s\n' % (item['cost'], item['currency'])
+        if item['date']:
+            result += 'Date: %s\n' % item['date'].strftime('%Y-%m-%d')
+
+        result += '\n%sShop:%s\n' % (self.BOLD, self.NC)
+        result += '\tName: %s\n' % item['shop'].name
+        if item['shop'].location:
+            result += '\tLocation: %s\n' % item['shop'].location
+        if item['shop'].info:
+            result += '\n\t' + html2text(item['shop'].info).replace('\n', '\n\t').strip()
+
+        return result
+
+
+class PricesFormatter(IFormatter):
+    MANDATORY_FIELDS = ('id', 'cost', 'currency', 'shop')
+
+    count = 0
+
+    def flush(self):
+        self.count = 0
+
+    def format_dict(self, item):
+        self.count += 1
+        if item['message']:
+            message = item['message']
+        else:
+            message = '%s (%s)' % (item['shop'].name, item['shop'].location)
+
+        if self.interactive:
+            backend = item['id'].split('@', 1)[1]
+            result = u'%s* (%d) %s%s - %s (%s)%s' % (self.BOLD, self.count, item['cost'], item['currency'], message, backend, self.NC)
+        else:
+            result = u'%s* (%s) %s%s - %s%s' % (self.BOLD, item['id'], item['cost'], item['currency'], message, self.NC)
+        if item['date']:
+            result += ' %s' % item['date'].strftime('%Y-%m-%d')
+        return result
+
+class Comparoob(ReplApplication):
+    APPNAME = 'comparoob'
+    VERSION = '0.c'
+    COPYRIGHT = 'Copyright(C) 2012 Romain Bignon'
+    DESCRIPTION = 'Console application to compare products.'
+    DEFAULT_FORMATTER = 'table'
+    EXTRA_FORMATTERS = {'prices':       PricesFormatter,
+                        'price':        PriceFormatter,
+                       }
+    COMMANDS_FORMATTERS = {'prices':    'prices',
+                           'info':      'price',
+                          }
+    CAPS = ICapPriceComparison
+
+    def do_prices(self, pattern):
+        products = []
+        for backend, product in self.do('search_products', pattern):
+            products.append(product)
+
+        if len(products) == 0:
+            print >>sys.stderr, 'Error: no product found with this pattern'
+            return 1
+        elif len(products) == 1:
+            product = products[0]
+        else:
+            print >>sys.stderr, 'Error: too many results, TODO'
+            return 1
+
+        self.change_path([u'prices'])
+        for backend, price in self.do('iter_prices', product):
+            self.add_object(price)
+            self.format(price)
+
+    def complete_info(self, text, line, *ignored):
+        args = line.split(' ')
+        if len(args) == 2:
+            return self._complete_object()
+
+    def do_info(self, _id):
+        if not _id:
+            print >>sys.stderr, 'This command takes an argument: %s' % self.get_command_help('info', short=True)
+            return 2
+
+        price = self.get_object(_id, 'get_price')
+        if not price:
+            print >>sys.stderr, 'Price not found: %s' %  _id
+            return 3
+        self.format(price)
+        self.flush()