Add support for Raiffeisen Bank

gcconv.py

@@ -262,7 +262,107 @@ class FioTransactionParser(BaseTransactionParser):
         return self.state
 
 
+class RaifTransaction(BaseTransaction):
+
+    # (a) Transaction Date
+    # (b) Booking Date
+    # (c) Account number
+    # (d) Account Name
+    # (e) Transaction Category
+    # (f) Accocunt Number
+    # (g) Name of Account
+    # (h) Transaction type
+    # (i) Message
+    # (j) Note
+    # (k) VS
+    # (l) KS
+    # (m) SS
+    # (n) Booked amount
+    # (o) Account Currency
+    # (p) Original Amount and Currency
+    # (q) Original Amount and Currency
+    # (r) Fee
+    # (s) Transaction ID
+
+    def __init__(self, line):
+        self._fields = [self._cleanup_field(f) for f in line.split(";")]
+        self.date = self._convert_date(self._fields[0])
+        amount = self._parse_currency(self._fields[13])
+        if amount >= 0:
+            self.amountd = amount
+            self.amountw = 0
+        else:
+            self.amountd = 0
+            self.amountw = -amount
+        self.currency = self._fields[14]
+        assert self.currency == 'CZK'
+
+    def _cleanup_field(self, field):
+        x = ' '.join(_unquote(field).strip().split())
+        return '' if x == '-' else x
+
+    def _description(self):
+        (
+            transaction_date,
+            booking_date,
+            account_number,
+            account_name,
+            transaction_category,
+            accocunt_number,
+            name_of_account,
+            transaction_type,
+            message,
+            note,
+            vs,
+            ks,
+            ss,
+            booked_amount,
+            account_currency,
+            original_amount_and_currency,
+            original_amount_and_currency,
+            fee,
+            transaction_id,
+            ) = self._fields
+        out = []
+        if message:
+            out.append(message)
+        if note and note != message:
+            out.append(note)
+        return " / ".join(out)
+
+    def _convert_date(self, text):
+        day, month, year = text.split('.')
+        return '-'.join([year, month, day])
+
+    def _parse_currency(self, text):
+        return float(text.replace(',', '.').replace(' ', ''))
+
+    def to_gc(self):
+        """Convert to GCTransaction"""
+        return GCTransaction(
+            date=self.date,
+            deposit=self.amountd,
+            withdrawal=self.amountw,
+            description=self._description()
+        )
+
+
+class RaifTransactionParser(BaseTransactionParser):
+
+    def parse_line(self, line):
+        return RaifTransaction(line)
+
+    def advance(self, line, lineno=None):
+        """Choose parser state according to current line and line no."""
+        if self.state.nxt:
+            return ParserState(ths=True, nxt=False)
+        if self.is_closed() and line.startswith(u'Transaction Date;Booking'):
+            return ParserState(ths=False, nxt=True)
+        return self.state
+
+
 _parsers = {
     'CzMbank': MbankTransactionParser,
     'CzFio': FioTransactionParser,
+    'CzRaif': RaifTransactionParser,
 }