From 0557a0e845d239e04224fa504533f4107cbe1a05 Mon Sep 17 00:00:00 2001
From: AjmalCybro <ajmal@cybrosys.in>
Date: Thu, 23 Oct 2025 17:04:11 +0530
Subject: [PATCH] Oct 16 [UPDT] :  'import_bank_statement_odoo'

---
 .../demo/demo_bank_statement.xlsx             | Bin 0 -> 5689 bytes
 .../wizard/import_bank_statement.py           | 159 +++++++-----------
 2 files changed, 57 insertions(+), 102 deletions(-)
 create mode 100644 import_bank_statement_odoo/demo/demo_bank_statement.xlsx

diff --git a/import_bank_statement_odoo/demo/demo_bank_statement.xlsx b/import_bank_statement_odoo/demo/demo_bank_statement.xlsx
new file mode 100644
index 0000000000000000000000000000000000000000..be15f27d791c20a67632b7b5d8dd486e52d64c6a
GIT binary patch
literal 5689
zcmaJ_by!sU)+Pjo?vjofQo0dA5a}2YB%~Qam;q*xP5}j^2T(YON=l<hcQ;akbc0BT
zfaEuL&U4^;zw6r1vuFRY*V^-2@ve6*Z8gjr6leed0PU8@Lu0gSf{%JOgF_q<{Ct=9
zg1AP_W&wg{Ti)T^hoVz*fOlEd>dMtDB8&~}FuKyP^p>FAgL~v;1d$(`qg?#_&n+{j
z)nUBT0w6WhFg?RCj9QSDZ+1)fxr-;iSZ|tN8>xhnj*|j6JX%Y2zh~}lSBAH1j4Ix@
zJq@)1^cAahYcv_T=Wr7}ccF9Z(B)vm^$0Q-Tm7`J{x@oZHJ}cnFcl_xuRFr?4_T2c
zZXK%X=|0&KHDrX4XLz2|5Z}t~nXPytS^@!O4bqn^%bOikL*-E{XYt*uuU?T?RW8Gw
zE%&i|Ysi8HY+b!I*nE}MH{Vb)2%D+CFiJh`UMAO8!vTaOizIKNoY04XhNk^L-9(If
z!^4r^-3jhs>Ez_V=K*z$)SZBj3y}D3J@nWos&nQNcts%U5tlR#%IGX{Ecy`6>zM(R
zlR7&rgrsB*-PuRasQtD+Hnp(STbVd5OvCdj+)#qx6%O}ki1x`VUvFg`;bp_c+*uu-
z&@n9DYQtLN)K8MuMohQ8iX3~mTWsb|#ERZ7yS=B=rr;{gk&LrHsDl!0H|e7psLA3z
z%`ro~Gk9Jsa1r#bTW7I&;O7VI5Qb3n&e*RLsv?O6iJ39mUqu&pvVQhBq`h`d688V0
zdXl#zl*$yPp*kkWQ~60QQvY}#ctRwpqx491nUbC*1!v*T5-}h$UIo%ICdG)AvcBcv
zAk|_FZF!;JvBW2C)b;9}q-h>0nBr$WYqgs;@$Hh@y;JrF@3aKdDNUv)wKk90@$&5g
zhXx!cPhfY~c_PMmQ<QoXQm4v;!pGR(+GYw+nUFqj%0b$`^(s&#{OKeGmtND-stl}p
zX`z<TSN)V*`ZaS%Y6xW#dz%dlr}<`h2g`xLFidlg8t`nF-1osf7Dq{1PTAF`%p<{W
zk@b8lkDcULIpUg`0ZmlYeMbXn(nUIP$(#P2D&akH9L(9kWroPmvFO7gyX2m%0cNA;
z>33;h96`kCJJvC<p?J_s$2iYpr0bG4^}d?aITfKV`=RzG-=_&gPuU9n;<(E$SHX$<
z@Tn<zp4nyIh$KfpIzdN6bH@Br-VpwsHwafxM+o9FYc>o9V1Efv_|7~u@G}b((HO;j
zU!|$;os=t@0MhVwZV1q`Vq5t1&aB=+9nk-2V$fbKbFua4DAk)7XF~93CGTmvbS^hY
zF`$->nvL5sNgx*M$cC}j_c=H-Od0U<DRxB`<sIUgql7}q+CHu?0HILshRTs8@AjPe
z6sioVI3T4!oT?rPp-F<IUJj=Uw*q&Cu*+1EE(yr_+Y1OOWtIGgM|gxnQh>2#1jejm
zc&vsy7kSKX7+x-*UGtkX^l86V=GR5ZMk>JBcl0SzB9+_ZjB0$8I%BL2;6*W6One<}
zIyV&wV@{ifc#4Qos#KB1l=pbBiDie}iJR%F>G9Ozh6QKAo2diTZgP1*QLBrl(kwyZ
zD3@uO3fwZeZ!hmT<lei%Fl_pHjV!e;VAXC&#HBv8k0PhGQzg1zMBAN34`Zm_Y)qch
zf*tvE`!Aset~tt9hJEn#3sIWml&qU%Jc&zLUp~63lNCVFsU4!Ypu$`F=%Q*wjG9Dh
zHYP9v`3*tqkca_j?l`^xFe}B3uBqI0Ap&1se|3S9E1AfVzHQ)O(B`pp2CpZB3Nj<f
zkm{&EnBXmbpAfhoJev@g+*URA_NK&srxeMO&IcT2XqMgx3}7QW#48q4>lq<pvRb4=
z{uh?W6pv0H`g(*dt36(`K>`m5wjyLXCGhoun-Iy%om(0stcXOkQtH+G!C2kRHV&tU
zZ;W1Br5`)b<op=+l(7Wi2x^6nc|sXl=>NIaS#7*(v@YUxor;Fl4^?}u-wdam*$Qf2
z9Kne9Y7wV9h3!Xr#|EBepYb#6%Y@Gw%2vtCJSU*us}B8psjX<*U&;wg@ejL;WK(SF
z>*d^p9LrT~)Xt!vQ`{;KOq;4RrJH3al8E`y&n@dhcT%A*fD6gE69`6fJ_tDGNsYS@
zGrg>e+{_dcgS{AYZ&_TTDwuSXVa2r}=}tTCu$TDkYuJJb7?S0qBuS^euJk4{sejC;
zhQJ@=vRL$SdoXrU0E<rfr(&VG0$GF|1mcR||MeqySvR#0H4y0%B(+;bZ!4bjV@Bzg
zWrN&4mXs!S4ueW`OT}q}!HUfKaU1OQenw>GtSJTbA9q*+_4iB<9dN5w-Z1ifUl*hf
z{(>*Qk1^W#qz$g)M7<e(vQ~l^B_;bJnU>#%e~62JIJ8^-GjIn7Y{|=j_P(|0R;DAL
zkC`>FFlO5z$iaY%rYI)ZxbBByvtZW`1<D`k%*tnYvE!lu>eE{^`JBqnMtLXnDiX1S
zt-{%Z`Beu)i8ciN$xZtP&~-NEX@e`ocw|gzqPMx@TtEgawGnzyy9Hj`^L52ky@SLC
z?iMDuURpHqn8|0_tZ!T+HsOVh^@wcd2bSjvOWH3nz=0M8ca43XFGb%k>h?ZhAej1C
zpK`L%-0i6;C$H!Be8F<gkBGleXXzcoIp~NPdo+r3Z0_d#1eje)0($xydHQUaCO>Y=
zXu)i~*m55qfG+J&u}zHKDuu~0LB?9s7_!a+9G|oZ7Y^H-LzGLa1TJpoj0=ooE>_4%
zbL#=*+C6M%6k71h_2D`J){$uj7OHy5g-EU$Sv1*5la-_o-}w|M?%@TQJ6~wX!i*DA
z#J`1YeB#~6l_H>PV~kgs!_HNf!UsA94L91SJ2TuEkr5j>r3EPjd5(}?d=G!$T4#yp
z<uOIMVzfoe>6VA}Z5z4x@lnN!DVb8qzEASL{9VI3bD&g<%+9Q}tb(MJnlYM9d|jir
zdZQVR$S3qbm45%m`jntt`3t>cone4pBbCSmZfCBTfNpx+qGX}g;Nw0#8Tt6_H^ayE
zF<QCzRiT*VnziOxH@wWc-e*Q=f;DT{ef_&s)FdVE(qc{X`uTTNWJYL#L8kk%+Mrty
zxa!vyHRNHox^dS&1?0O%?@uAR(VG3(d)<<@IM3HKY(sk)w<@FB+wqz_F4{c@Pt*6A
z9Tc;t)#Xc?nlbWEAOeSn+zlfO@O9F<eS=7uw3m!)AKDi)+Y@#|KZ!9b7AfsFdfk(X
zGAFTjTov4ZU#djb@OjWW0%?#OA)}*){6?t(V#@EdrdS;wwB7}2v`MTaf~<&*k)1rr
zEll|;x5Hg+?8)j?kb;_amHF*_Wt=-aY3xkAszRD=UJoCE*!229L&M`ZdE{~jJdkzE
zyttY))#Jw8j0Nlb@)?NKG!MrENp$Pllo2ax<Ah=TRwkd|o0^Y3C%!edalJ!6vsnD<
z=@e~J4h`I_zfcy^Of|ekYSQ>A_~$GVkL?Znm`9^HTg$_i@*AN!o4uG$<8Eskq9P^7
zc;Blm$hZfm5~`LSx~tHn;;398UOyG|^y=JdPc*@tf(QL%@O&2igJ7T5bs1|;KKJ}&
zqHx@feKbM>YDH33?)zd*3hy9sEfntdeoO7qjQnyjL?W=|$JYvHx;CKF4X-^y#j)lU
zsS880F)a-ug0LW5V9<TvR&G&$JW=am^9fuZsM08OsSbPXSp$v&WH8MClXX6ef&a_9
z1Z?xABi)@UV7b7@Ik-`RPnIWc*N-!4b-?>}Ckaf8-(ZhA+ml3A)-%R6LS;w!{n0Ox
ztXoC3KKKS28Xd`>knHB)NcK0ezD^qdCfTtX4!=nD8Rh-MRuvqp2#MN=yG3V5=xyNn
z{uZ&gk<tr4K00GO$a+7re{`=-`NwyW$nT*o3Iy1>X*WU|TipP)foGc=bF})sY16ZI
z!02Kovb-AduL0C5{6-~v5{7)1O)=6Y8J(({IisCbK^tKH?p>9k6!cz7-2!9GPcR<s
zYDCjTvDGLcky*O;8-Wf&++H|itnh$NwB;-hzYxwo^yiYyk}NC3SGKOApf@^V?zmsf
zY#6t^@SSic3bg&!Pucf&%aarfil(Q0eqp@U_D`=t)VYCwB+?ZB5F$GZIK*1l6>bl+
z{l$wCkiDqEgCuzCcF2<VV{zGA1@sRA!7W%W0XW5RnI9;Mf<|)!vfCpioaOrl4tm5(
zcnn|Pnd;yp>F{#=>Bc6Bx$fh%Gr3zn+CHBs9FFZR>=m5okG_Mf1K*1U^5}Fwr^!wy
z877=1fS%rXWjgKa49c>Sz)!|_m5Lqk?LJtwlDyAgF@I0Ejlz4@@&1Wm^VybSb17@y
z7czK8rz@8<7|?`OQMX3L;VniKA8HhqcO)BB{k<fE7_=;CrXC7lslR9P<s`Q&F7cxy
zo+qbjlKAu?eSnNbu(@vszitsvNiT=^S_kBgXL(dg`Bq4cnbLX5?by1Y^IutI9ICi-
zjPi{7AIT&s>J4irD{Z)wGlJjB2@d%+hUl1R1)&J#;lr3Z1$=9SeE{%r;wuTO`G~f1
zqfG-!)gd>llS3*R$M1|j)hnq(i2CH7L%r1vIXZARMF{#`MrG3#c`7b1!&|4cyAco}
zTgWXIMH~UUR1@GMu&IN`U?~O^i{>+}ibAF8K%0@GpqLLPrEs9GvQnXgiQw8>zGBPM
z7nA{P@>vxa-EbRP7@@DIR9ZTuPf4xx3&b`CK>&QfT+nDiGLIa2TG-Qy!+^-j50>Gq
z;AH3k*NR(Gh-r|ExNtl<dwvA{xjh;A9e=Da9J4FViY?y3oC^M#w!dz&p{A|eJrApk
zl}zI_=E%B?RalBeB)N{`X!K;HINITL={jzO1Wt3Q_Q+H*hd(<#UlMN_Y>rm>;`{?L
zBIh6;DD1WxRYw&wL!Z(fGT9G)(sO2&!D$*o3KWbi{Cxi=K~>jHx3n{F!%0d8i;2=P
zKOc<pGi5gC50quWcVvS7&#ev!FN=Tm3I2f{it|tWF~w2Ha1|yCXXnc(y^hh+LX~{$
zmQOhMc&Wz2Tp<m65Xrm<E)kYX-p7`yr@LBV($JYyw81iAaaq|l+mjEyoyqe6LHjMQ
zkNJ;v0+6Qm<y)O6+d75HvDy;A2={)ZF#E?j_w`%HXr!VUtmKcrccKgCb7}0N>1S|H
z@JS=1_01Dqw9Z8AaX*{(QPB@(6-wiA+{l(on8RTAZ}~xbfv;zJKKb&_GepPjKJ4N~
zz>U6drS!M)gzfN8{gO_oTNxDujPQD5w+p5fJ<g`?R_7Kl=GuJt0=pCWCJ(kgK5t{*
zBkL1*Z+zrMwjrIKSj2P_O>J9S{jY@1nW~>&NBO}U<*olcRg+)$l+_c2s}uCnS*1f7
zPEE`-INK+86q{@1^Yk3XZ$d?-*Na<TN~aIUmCB(_B|~O>T}!=0#^Q&X)y`#^NqS}1
zx+KAwNwb36*Z_%m@>pTN&_-CbTXw?$_8$Gir$aO%4}v^*DQdY{a)crtw}YG+f(B_M
zgS3%xBL>9lBIUv%+M;4*c|_V2SZpHXiw3%|x6%DHap_`dZc9KSSI-jUPX}v0<!b5L
z8L`1-;DBU}OZa&4ZEQ2c7#qU?j8U_YoHZ7w)o0Dl^J!f2=Yv|KbCy2{>BsmE0-If+
zT6Hn-nJHxC?Ic5>>?d&83ULgSw1!BrnEJw=*n5J}S)mgwIOHlp2V}5+QSFkR>Rm>i
z&D}Qh#K_X`t>^PPT=46$nO&H3e=~nUCDJ(7LV+nF%G*`JSy7knhq*HxT`hlH+<PKu
z)G6^4<!~H;tl+kS9Lo85e?*ka02nJc!CWCQSF;D6&JcvjWdLd)^lCLTQ`B!2*;OPr
zEjQ8(!Ej`ws#-k707bV7eQtc6JokS01(H;XFkA?0@UG`KW&R6l{W`x<>YZK`$RmPG
zTOIMPi@uhZanM2Juc(i$BLQmiMR3X78gcPp3H`tfA_q+F2Y6864+{r1OHSiKvW_JW
zAH`StjPY8&i*iHjM0i!XzVLe7_5xTy`eNC}3mwU{w&AB>rY?4U6^=gn4X*%)R^rJ_
zvp#1zF_}WLHj~pr^9P>H%+~5upQ}l?u&Y=!Sj<zKo)|Z>B<Zd(<q!}qih@`?W^bFQ
z{*~-wSM4+86?RfotDr$0V%{KX+=<rMXYh7X56yI>!tg_eJi~yypK%bKQ7y}v!Y=l#
za)~;&*oOt#`E-WEbNX}p{lYY=qmuGDU*bD~zMFTL&}GZVE4GCVB&J&2H$ycwm@(EA
zFTS~l2Oh5vlgvor_~nWjOBOAS?cVCu+5z8BHuE~hy$sU4bUCp#DoDdvmsNs}L4kI)
zq`SVCx?Ixzr(9dv{jPX@F>$%Jx)Ki5r+!;v{jPj{P`lh+T!|Sfkx<H4n~dL8uQQ{|
zN$yG_u>bw;f6aBjTe;36E?Lx-=%c)V`UC%iQ~hq|dPlwF99QB@@b^J}XCS{@xDNK0
zo%c%eQJ(p)KK#4-bp*WZ2v?$uipsz0|MZ97wXc`&WszQqFUr`z+E<wQyMgPe^`8dL
bQHS||YF=9n3l%0bG(6N5fa-7QzqI}jp-^Wa

literal 0
HcmV?d00001

diff --git a/import_bank_statement_odoo/wizard/import_bank_statement.py b/import_bank_statement_odoo/wizard/import_bank_statement.py
index e3c42860b..36a06aa19 100644
--- a/import_bank_statement_odoo/wizard/import_bank_statement.py
+++ b/import_bank_statement_odoo/wizard/import_bank_statement.py
@@ -47,76 +47,55 @@ class ImportBankStatement(models.TransientModel):
         """Function to import csv, xlsx, ofx and qif file format"""
         split_tup = os.path.splitext(self.file_name)
         if split_tup[1] in ['.csv', '.xlsx', '.ofx', '.qif']:
+            # ----------------------- CSV Import -----------------------
             if split_tup[1] == '.csv':
-                # Reading csv file
                 try:
                     file = base64.b64decode(self.attachment)
                     file_string = file.decode('utf-8')
                     file_string = file_string.split('\n')
                 except:
                     raise ValidationError(_("Choose correct file"))
-                # Skipping the first line
                 firstline = True
                 for file_item in file_string:
                     if firstline:
                         firstline = False
                         continue
-                    # Reading the content from csv file
                     if file_item.split(',') != ['']:
-                        if file_item.split(',')[0] and file_item.split(',')[
-                            1] and file_item.split(',')[4]:
-                            date_obj = str(fields.date.today()) if not \
-                                file_item.split(',')[3] else \
-                            file_item.split(',')[3]
+                        cols = file_item.split(',')
+                        # Expected columns:
+                        # Reference, Amount, Date, Partner Name, Start Bal, End Bal
+                        if len(cols) < 6:
+                            continue
+                        ref, amount, date_str, partner_name, start_bal, end_bal = cols[
+                                                                                  :6]
+                        if ref and amount and partner_name:
+                            date_obj = str(
+                                fields.date.today()) if not date_str else date_str
                             transaction_date = datetime.strptime(date_obj,
                                                                  "%Y-%m-%d")
                             partner = self.env['res.partner'].search(
-                                [('name', '=', file_item.split(',')[4])])
-                            # Creating a record in account.bank.statement model
+                                [('name', '=', partner_name)])
+                            start_balance = float(
+                                start_bal) if start_bal else 0.0
+                            end_balance = float(end_bal) if end_bal else 0.0
                             if partner:
                                 statement = self.env[
                                     'account.bank.statement'].create({
-                                    'name': file_item.split(',')[0],
+                                    'name': ref,
+                                    'balance_start': start_balance,
+                                    'balance_end_real': end_balance,
                                     'line_ids': [
                                         (0, 0, {
                                             'date': transaction_date,
                                             'payment_ref': 'csv file',
                                             'partner_id': partner.id,
                                             'journal_id': self.journal_id.id,
-                                            'amount': file_item.split(',')[1],
-                                            'amount_currency':
-                                                file_item.split(',')[2],
+                                            'amount': amount,
                                         }),
                                     ],
                                 })
                             else:
                                 raise ValidationError(_("Partner not exist"))
-                        else:
-                            if not file_item.split(',')[0]:
-                                raise ValidationError(
-                                    _("Account name is not set"))
-                            elif not file_item.split(',')[1]:
-                                raise ValidationError(_("Amount is not set"))
-                            elif not file_item.split(',')[4]:
-                                date_obj = str(fields.date.today()) if not \
-                                    file_item.split(',')[3] else \
-                                    file_item.split(',')[3]
-                                transaction_date = datetime.strptime(date_obj,
-                                                                     "%Y-%m-%d")
-                                statement = self.env[
-                                    'account.bank.statement'].create({
-                                    'name': file_item.split(',')[0],
-                                    'line_ids': [
-                                        (0, 0, {
-                                            'date': transaction_date,
-                                            'payment_ref': 'csv file',
-                                            'journal_id': self.journal_id.id,
-                                            'amount': file_item.split(',')[1],
-                                            'amount_currency':
-                                                file_item.split(',')[2],
-                                        }),
-                                    ],
-                                })
                 return {
                     'type': 'ir.actions.act_window',
                     'name': 'Statements',
@@ -125,8 +104,8 @@ class ImportBankStatement(models.TransientModel):
                     'res_id': statement.id,
                 }
 
+            # ----------------------- XLSX Import -----------------------
             elif split_tup[1] == '.xlsx':
-                # Reading xlsx file
                 try:
                     order = openpyxl.load_workbook(
                         filename=BytesIO(base64.b64decode(self.attachment)))
@@ -134,81 +113,52 @@ class ImportBankStatement(models.TransientModel):
                 except:
                     raise ValidationError(_("Choose correct file"))
 
-                for record in xl_order.iter_rows(min_row=2, max_row=None,
-                                                 min_col=None, max_col=None,
-                                                 values_only=True):
+                for record in xl_order.iter_rows(min_row=2, values_only=True):
+                    # Expected columns:
+                    # Reference, Amount, Date, Partner Name, Start Bal, End Bal
                     line = list(record)
-                    # Reading the content from file
-                    if line[0] and line[1] and line[3]:
-                        partner = self.env['res.partner'].search(
-                            [('name', '=', line[3])])
+                    ref, amount, date_val, partner_name, start_bal, end_bal = \
+                        line + [None] * (6 - len(line))  # ensure 6 items
 
-                        if not line[2]:
+                    if ref and amount and partner_name:
+                        partner = self.env['res.partner'].search(
+                            [('name', '=', partner_name)])
+                        # Handle date formats
+                        if not date_val:
                             date_obj = fields.Date.today()
                         else:
-                            if isinstance(line[2], str):
-                                # Try multiple date formats (e.g. 2025-09-20 or 20/09/2025)
+                            if isinstance(date_val, str):
                                 try:
-                                    date_obj = datetime.strptime(line[2],
+                                    date_obj = datetime.strptime(date_val,
                                                                  "%Y-%m-%d").date()
                                 except ValueError:
-                                    date_obj = datetime.strptime(line[2],
+                                    date_obj = datetime.strptime(date_val,
                                                                  "%d/%m/%Y").date()
                             else:
-                                # If it's datetime/datetime64
-                                date_obj = line[2] if isinstance(line[2],
-                                                                 date) else \
-                                    line[2].date()
+                                date_obj = date_val if isinstance(date_val,
+                                                                  date) else date_val.date()
+
+                        start_balance = float(start_bal) if start_bal else 0.0
+                        end_balance = float(end_bal) if end_bal else 0.0
 
-                        # Creating record
                         if partner:
                             statement = self.env[
                                 'account.bank.statement'].create({
-                                'name': line[0],
+                                'name': ref,
+                                'balance_start': start_balance,
+                                'balance_end_real': end_balance,
                                 'line_ids': [
                                     (0, 0, {
                                         'date': date_obj,
                                         'payment_ref': 'xlsx file',
                                         'partner_id': partner.id,
                                         'journal_id': self.journal_id.id,
-                                        'amount': line[1],
+                                        'amount': amount,
                                     }),
                                 ],
                             })
                         else:
                             raise ValidationError(_("Partner not exist"))
-                    else:
-                        if not line[0]:
-                            raise ValidationError(_("Account name is not set"))
-                        elif not line[1]:
-                            raise ValidationError(_("Amount is not set"))
-                        elif not line[3]:
-                            if not line[2]:
-                                date_obj = fields.Date.today()
-                            else:
-                                if isinstance(line[2], str):
-                                    try:
-                                        date_obj = datetime.strptime(line[2],
-                                                                     "%Y-%m-%d").date()
-                                    except ValueError:
-                                        date_obj = datetime.strptime(line[2],
-                                                                     "%d/%m/%Y").date()
-                                else:
-                                    date_obj = line[2] if isinstance(line[2],
-                                                                     date) else \
-                                        line[2].date()
-                            statement = self.env[
-                                'account.bank.statement'].create({
-                                'name': line[0],
-                                'line_ids': [
-                                    (0, 0, {
-                                        'date': date_obj,
-                                        'payment_ref': 'xlsx file',
-                                        'journal_id': self.journal_id.id,
-                                        'amount': line[1],
-                                    }),
-                                ],
-                            })
                 return {
                     'type': 'ir.actions.act_window',
                     'name': 'Statements',
@@ -217,8 +167,8 @@ class ImportBankStatement(models.TransientModel):
                     'res_id': statement.id,
                 }
 
+            # ----------------------- OFX Import -----------------------
             elif split_tup[1] == '.ofx':
-                # Searching the path of the file
                 file_attachment = self.env["ir.attachment"].search(
                     ['|', ('res_field', '!=', False),
                      ('res_field', '=', False),
@@ -227,20 +177,19 @@ class ImportBankStatement(models.TransientModel):
                     limit=1)
                 file_path = file_attachment._full_path(
                     file_attachment.store_fname)
-                # Parsing the file
                 try:
                     with codecs.open(file_path) as fileobj:
                         ofx_file = OfxParser.parse(fileobj)
                 except:
                     raise ValidationError(_("Wrong file format"))
-                if not ofx_file.account:
-                    raise ValidationError(
-                        _("No account information found in OFX file."))
-                if not ofx_file.account.statement:
+                if not ofx_file.account or not ofx_file.account.statement:
                     raise ValidationError(
-                        _("No statement information found in OFX file."))
+                        _("OFX file missing account or statement info"))
 
                 statement_list = []
+                start_balance = ofx_file.account.statement.balance or 0.0
+                end_balance = ofx_file.account.statement.balance_end or 0.0
+
                 for transaction in ofx_file.account.statement.transactions:
                     if transaction.type in ["debit",
                                             "credit"] and transaction.amount != 0:
@@ -258,6 +207,8 @@ class ImportBankStatement(models.TransientModel):
                     for item in statement_list:
                         statement = self.env['account.bank.statement'].create({
                             'name': ofx_file.account.routing_number,
+                            'balance_start': start_balance,
+                            'balance_end_real': end_balance,
                             'line_ids': [
                                 (0, 0, {
                                     'date': item[2],
@@ -278,8 +229,8 @@ class ImportBankStatement(models.TransientModel):
                 else:
                     raise ValidationError(_("There is no data to import"))
 
+            # ----------------------- QIF Import -----------------------
             elif split_tup[1] == '.qif':
-                # Searching the path of qif file
                 file_attachment = self.env["ir.attachment"].search(
                     ['|', ('res_field', '!=', False),
                      ('res_field', '=', False),
@@ -288,7 +239,6 @@ class ImportBankStatement(models.TransientModel):
                     limit=1)
                 file_path = file_attachment._full_path(
                     file_attachment.store_fname)
-                # Parsing the qif file
                 try:
                     parser = QifParser()
                     with open(file_path, 'r') as qiffile:
@@ -296,13 +246,16 @@ class ImportBankStatement(models.TransientModel):
                 except:
                     raise ValidationError(_("Wrong file format"))
 
+                statement_list = []
+                start_balance = 0.0
+                end_balance = 0.0
+
                 file_string = str(qif)
                 file_item = file_string.split('^')
                 file_item[-1] = file_item[-1].rstrip('\n')
                 if file_item[-1] == '':
                     file_item.pop()
 
-                statement_list = []
                 for item in file_item:
                     if not item.startswith('!Type:Bank'):
                         item = '!Type:Bank' + item
@@ -326,6 +279,8 @@ class ImportBankStatement(models.TransientModel):
                     for item in statement_list:
                         statement = self.env['account.bank.statement'].create({
                             'name': item[0],
+                            'balance_start': start_balance,
+                            'balance_end_real': end_balance,
                             'line_ids': [
                                 (0, 0, {
                                     'date': item[2],