diff --git a/src/documents/tasks.py b/src/documents/tasks.py index 7fde919fc..7bd212399 100644 --- a/src/documents/tasks.py +++ b/src/documents/tasks.py @@ -198,7 +198,10 @@ def consume_file( for n, document in enumerate(document_list): # save to consumption dir # rename it to the original filename with number prefix - newname = f"{str(n)}_" + override_filename + if override_filename: + newname = f"{str(n)}_" + override_filename + else: + newname = None save_to_dir(document, newname=newname) # if we got here, the document was successfully split # and can safely be deleted @@ -214,10 +217,14 @@ def consume_file( "status": "SUCCESS", "message": "finished", } - async_to_sync(get_channel_layer().group_send)( - "status_updates", - {"type": "status_update", "data": payload}, - ) + try: + async_to_sync(get_channel_layer().group_send)( + "status_updates", + {"type": "status_update", "data": payload}, + ) + except OSError as e: + logger.warning("OSError. It could be, the broker cannot be reached.") + logger.warning(str(e)) return "File successfully split" # continue with consumption if no barcode was found diff --git a/src/documents/tests/samples/barcodes/barcode-128-PATCHT.png b/src/documents/tests/samples/barcodes/barcode-128-PATCHT.png new file mode 100644 index 000000000..80517d56d Binary files /dev/null and b/src/documents/tests/samples/barcodes/barcode-128-PATCHT.png differ diff --git a/src/documents/tests/samples/barcodes/barcode-128-custom.pdf b/src/documents/tests/samples/barcodes/barcode-128-custom.pdf new file mode 100644 index 000000000..f603dff5f Binary files /dev/null and b/src/documents/tests/samples/barcodes/barcode-128-custom.pdf differ diff --git a/src/documents/tests/samples/barcodes/barcode-128-custom.png b/src/documents/tests/samples/barcodes/barcode-128-custom.png new file mode 100644 index 000000000..c3f1b803a Binary files /dev/null and b/src/documents/tests/samples/barcodes/barcode-128-custom.png differ diff --git a/src/documents/tests/samples/barcodes/barcode-39-PATCHT-distorsion.png b/src/documents/tests/samples/barcodes/barcode-39-PATCHT-distorsion.png new file mode 100644 index 000000000..3f858f6ad Binary files /dev/null and b/src/documents/tests/samples/barcodes/barcode-39-PATCHT-distorsion.png differ diff --git a/src/documents/tests/samples/barcodes/barcode-39-PATCHT-distorsion2.png b/src/documents/tests/samples/barcodes/barcode-39-PATCHT-distorsion2.png new file mode 100644 index 000000000..cc81f8e36 Binary files /dev/null and b/src/documents/tests/samples/barcodes/barcode-39-PATCHT-distorsion2.png differ diff --git a/src/documents/tests/samples/barcodes/barcode-39-PATCHT-unreadable.png b/src/documents/tests/samples/barcodes/barcode-39-PATCHT-unreadable.png new file mode 100644 index 000000000..1e24b4d84 Binary files /dev/null and b/src/documents/tests/samples/barcodes/barcode-39-PATCHT-unreadable.png differ diff --git a/src/documents/tests/samples/barcodes/barcode-39-PATCHT.png b/src/documents/tests/samples/barcodes/barcode-39-PATCHT.png new file mode 100644 index 000000000..0078026c8 Binary files /dev/null and b/src/documents/tests/samples/barcodes/barcode-39-PATCHT.png differ diff --git a/src/documents/tests/samples/barcodes/barcode-39-custom.pdf b/src/documents/tests/samples/barcodes/barcode-39-custom.pdf new file mode 100644 index 000000000..ca78b93be --- /dev/null +++ b/src/documents/tests/samples/barcodes/barcode-39-custom.pdf @@ -0,0 +1,243 @@ +%PDF-1.6 +%äüöß +2 0 obj +<> +stream +x]Qj0 +t.+Vl1=:m_ަb<xH1\?Ը*"uW8 D?%L Ztأǐ;l7p6s!3. +A-:k<⟿2ΤZj}3z(L zB -#.B<2sHk& +NYRDZ~'6:-%.[';n+ٜ%DQYV7Nun"g#`Rmx +endstream +endobj + +3 0 obj +284 +endobj + +4 0 obj +<> +stream +xn`IdY̢`A8+Mei~2jޭvҶؒƶի$zmIOw4̟75ݲN[ضz5DV/-4fC287o޼y͛7o޼y͛7o޼y͛7o޼y͛7o޼y͛7o޼y͛7o޼y͛7o޼y͛7o޼y͛7o޼y͛7o޼y͛7o޼y͛7o;g +endstream +endobj + +5 0 obj +11207 +endobj + +7 0 obj +<> +stream +xzy\U=UA7BoС + KQ!lI0tD: 4CwqqAQ22sflLEQtt9>͸ddqy&dԧNݪf +s9ls {@KP B~CX:ƹ5MU؞ ;ӄ=C!ٿ *YR< ,^~/"uw"{?C%/_J_wqrI.Tꈟ={=C^EtvxArH䜗N_͛O(ˣMXJ^POAG$#Ov޹X19з] d#vH#1ݰ9Qσ3kl&=d/^s-27 +|#}s({T_ m "1;7]xugBm%i%)eʉQ)>.$uz2 476$U kT/~THM/ԛDQc{y7mP_W[zPvU++W,_VW\T(/w!8f1=Mj*R% X`s9s]C E\ &*oh >\"_9`B@0SK5%>h[G;izJ%I& QFӧׇS +Xzl%`kW1D&Mֆ-m5.7PT!!Lh'%+Nn +LJou4lM-Eڪxn˘5oo{܄uu-m0VXnHP (u8#qڊw=.2DGU@'P. ya&y<:@Q#B[0 0zdBZ )Ic\c AXǦ9f4p!V +>Cc)KB A٩[3OZ0\=lVj.)PFkMZDcHꄞUJk$CC{%e6}$y*Ie~Z, ` jI gtn :Hau3)!L{X`Ä!;CQœvx#t,j&;x O^:dJ(ûttK(d4G&;D$'s4CtE%pڝkYHcLʜi.sw^Xf\[7{b((е^uwKQRn}VϿuB*Xr{ Rbfd=Y@6bǑ Q[jӈ&<0*sQ=paJ+2ၑ9gEwԺLjQ3XuUJ6T\F9U*Pmת˾3(^|;/vJzkn,*dvfOeH-5,…SEHxZG&lvٔaIJ6flFh6T<`:H8`v9@0I!:s~H&hVq?r~Χ0)kehvV˅BA(,R(fջr_j'eJke<'ܫfaH,^e?rX?eCK K%h_MK )/A# ABg$\TQ気V:l<>Gcnym'u?*)Zg|K*U~=$4YUziRܞ ?BgɔcKcύ\BtBF(9 L\ ϙ-i@H1 V4rs1]}&.~!z޽=0(=pYES?[|h:zF6>s/9 lqUÙH+FՕilֵ%ҚXF!%]i@]ݜҀ}p̺\g0dXҺ v,)Oښە,Y5SuBIᖔ@ny嚂\GmPZpNU0d3l H>·|C0|ȇ{)~ +|%e:lʇIʂPWa)g2Whrd.x+ÚSm`-S˖phݘT\; ??~st S7]/ſO(AVG> '=+܁qg%["V ͮ1̤,c5QkHLjku>*t;P-\a5DC + +3,XCMe<V桝U<AE_c:Ai^pA滐~1-|rʠ7~ɘIŚ=_x87sm5=5o:kn_ӹ5{z5nuE+pG7m[S[oyi~S76;Y&Ac6N+fUW{;/ݫ!F1!ψ fDȁ\̨x?0"DGSW +tܼXVKYi'55Ns^aѳ pA4 d U)*HO +&Ss^ew0['@rɋQ ;04::5q\uW/*=SU/+!b$h>5*K UZͰӃ &ZeЪ`l޳Vx +ǭp +Zj+ĭe+X + ` +So!8V`Va+ Y!j+X +,&+{Yx^&2 /-g L zY`X LM^ 7|UH+)JB3GƢcFd@Xk3\i.=j4*g \)LLv@ +kJKHe1q +QN}0)DTf$ERfbSo;ʬ?3ypMⵐxLJFCO^󋼩tU0pő+~q5}FJ!(VfMsf:Hé5b& \&LdH&D3g9&Osr&/?O}1E3rQ8&TtD@Y]:k&ن`10E`d`R岢;=ನi؉lɆh6A,/-9`iIHƗ0ܻ [^MQ\" x@:WeŧL:="v7uOx +Yn5ZPoFc'^ϰTӤ +5VCo(v)f+4,b +|~y&Y(^c3SmgyMo2"גF,ϵϙD|22>Z੊i.ogOה ~6ᵏm^e[uV쨛Un7})R];g(5=]ʠ71 荘[|q/=EW9TRr/K~ƷT/j< + 0[x,˯WwJV6W[w1-z㍖M*WۍN<[+-wVttZXKΟAˣ'j$l6iȱo/I3UCwކ-_U\m6YoDU8*@V*eݬ=6iܩƵ>Hm*z7OΤmM;6R 9`}%¤Q7 mF@mJ!ZҭYCJ[8)myJ[J[K.#ǕXXitVک0~' fcyKi +VIF^%({©X͐tY B]J[MT*m q~i#cJ;,PSwԟ+m=Y{Mi)zF.JIΕNS^,u!3twWO\̕,-#]}an}d0 {#ũF+"P0Yq`o=}u@Wx+8-<:KKfq!.> +/"ûzc {b\hk!l S`gx0BH%h`oS-V<k4{yx< Tb8J;rzz;{}K4P^d7\rwc=B{BqIp|3w)?T}?6q[#eQ6hS?:Ke,u8Y+Gn=P'Z ACp@QH4_8֌E@8%͈b !N\,DA=Es$ đ4…PqVsO'4s<)\s0cѾPNJ{Jo߾}!5bѰAK_#@r_I \SSq +B!̥K)Ќx8Wkk$5;e$Lw!luK A(Gߥ,ś#>zl" Q2i4|;RlmUhԅڀȡ:pt._4SH/YrكrĐ*8]#8ym)E]rNķ9qq:"IOa6{pދH(Ww+b4S,?, Ps؄3v#}'d"BvbӋރT.J-3玍f*^:y.ct1E/f-TB%[CIy{h;DEP(;0oShC_{)%BO9j~SNϥ*Gsu(h]=/G6{ۀ7x*rt+qQ(#To$MTR+)ܲl=4:BԷaqA^]Q +)"4.Vlz>sr-876%QycsxPift-a)3|ti܊6qe d˱A=z95˅}# +]f"K?]=4d%>N)q8wt*kXR ]3Yu{ߤ'15|UNwi՜3Ҝ9_ 9{Ĩ-p gh$J-NHR: Lz` NZaYzKbXCo b5Wa +&[(A }JH2> +{#]^Z&Ab^EOJ8SpS+Cdrɧ'?~cc0~ :#G|I5 4XN+9,9?9t2qR}wX4΍GLJ_ =u)O<'=O2cMcAσ]xnG=Gݞo[䙸m6uOB45hilpeħoMxG>7y݃lA悛/k9| ;tOebŞ@g~'٪-c[58 .lͯ vDڱij~'ҪFUhd=lFCӬVl{?gi5''&/r8PPc_zM{=܋u=] +u}uB[j/j*32.#>㴑1ۍTfj8Z +6Noݔw$DnlOhK;)pAR)Qܖf6%!H!l: ރOR;c2̌0E(HrY !@#C,$ŽpTQlV +endstream +endobj + +8 0 obj +8993 +endobj + +9 0 obj +<> +endobj + +10 0 obj +<> +stream +x]n0E|"RH$}@!E*9d3R5㸬sG,Y 3\zDNQra]7y: Q-]lºM f!穝brkyY˟}@$VŌSkQ.e!ө{d˹3RRfov)YoS8֐w^"5GX'kár%3dΟFq+̟Hפy9Fuί?2ט_q )ίꞟzV]<[Mq ޙQ s>.';{8o@g +endstream +endobj + +11 0 obj +<> +endobj + +12 0 obj +<> +stream +x8kxוΌ~J%ȃm9~1x001_H$#- I!miPh>iCڦiKxLдM ]HMڔmMS.I$ܫ14͏ߎ4s׽wd|40<_Y\.$Ͽ7)>ӂEPlȗOm t9o[dCA_ /j +!$ >C|KB#~/[8pﻹpI>g/;͟q1 >8q-"ĢKHIʏŃ1yd%OGFClxA32sr ]uֈEښvUD TH$+*8CIiIy^fhSHrMɖ7‡$[,T0$5W[]q M{G2NZ̬,RRl8pUSsi/t۬vf+1c&:ELa:#N՜w;)VxNw+jJjSn#*5R[RMvyuJhʍ86`8[S)r@A[^^Vzb% ozv|$X;hbz=VuG1zCdG ۩lu+\K ~n{l`r0v;M=2@DqqvXO(rq,3>ǙWJX>"oH{|]0Q}j&LJɊՆ@XT4$Z}CU& }'=e *LyJ P;RW QnMnEnC@kwϋ b*N)R}n)u +xlgJl]S8{qjh} +.hpu7$vE`=;mhea'ulqߨ:fPsByuf$5mPїE7g=(hDB@jmƧ+mĄ3*mfM0'n(Ub{M5F5uYR=U[![T'F =Mj )d?u0emz-%McaYVrj5؂da9&SqU[&WYy:98:&qI5h 7l/ Z½W4f zbJbF ̤q?z;+:IZ$rgJ&#W<}#:ogj OȨR"EDPK虼)`qF`h\fLOT&Csi8k +hL3l.NJ:xJfx<j24ʐiq:uـj ۥ(c] F'4XHkLtDdJV%KjJoIӵ%DZ +m%)u7 jѹ6Ҁ<&h9 z / )\ij45.Ϸ&|F5M +o_s]h+ +w=V`(¼ ]6[Al3sf {,h4pѠG{Nɸk'v"I=v20GZMyY=}l!s}V.'f`1ku,0(afW,̴pwdu𻏧>o,{zS&ȁOI>{́'ob$yq;ͩwb4An +heIeNg57-:z=Z-xrù\m. \.C,C3M `l &hA#680Gü6Ǩ1x_D"2XDEXKZMhiii^deRbi\56Zh` OLJv.- ͞SZÛߚyC7b֢Z%fKf3,|I6˓5krxAYkRL &#`J 5uD*3r\;+$/ϼ+~uBp)1xLʉd3>r >݇e᯴zĤ-d<F b[H|^wcvH, +~Jͤ.$~fᎽJSMHj NӏK}e 9+&yy33ʤ<.Zz-KL"N%wԈH81檱s!a=i97 +Y?M|H>h|OBv>? R˽'=4SOx>o<(0h*yS&o-)؍C`iތo$FVA+.3m"׮-gٲpW|%.ɼ;37CHC؉@L=7_;u&>v睴GoUσ>'{uZ"KY6@YqV[uCh2&vd{T];K5iJlYn1ZJPzk+|0vS;7y>c$'/6n?Izo%$Jt"L6/_b'=eN-քBoϵEl+bҡ;!gE..d41ܲucIW?[fyjc~%k+JVlXg}Fs#7ɋ3s9PPjZ|75Υ5jW4Z-zjCԡ/g[Cyw>T3?} ԳPVRN_wTflca o6flz pO:&A^q;ȱ9 AA.:9Q七;Ug⬇ +Qu(]H_=h,C3N^=wKvŗx8Ca 7H?614߱9ZMbx題u}7iZ!rp11e.>vư21rUw;[F2'bD <@X0Ý*}AnS*3YKU8I +g"븗U8x +B zB U@@T`Sa *,Cت(Ta-T_Ua\~zphNp,Ҽ™ T8 nԟUlؖQؕS\X\[xg8=OGcᝡW Kᠸ.Ed8\wX؋&:|qC_1#|Dkt86FX+^ǽG%4-*@@'}b2 G|btZWxpg8 Ƒu}ub/$E_$ nW +Ɠ>&Cx8l8$/o%D4K\xx$Z#8K`"3kuD0H$M ֠C`"4-`<<!_F>L~>,H Uw`7|.eS*Gb^^m#8/'F1Y?9cHmh< o*n &t$ $h!0*nP4E#IT@cDE#Ds><ذ/VFud2UcQвx}Z822+UeAm(v0?.tNTjĹ\ZT%up]4m!;Q@ۇ!?D! ΤBHUU86@=,[(E0ꋸiGQ>ƞ>f7 +Lpk ^t06-lD]hW>"VćQs-~ LeEu?+IK4WK@ sLvZD܁'~uՍ`~ҳP[Q^*{5nRTdQ\jYEA|l@a6Oڏ hPpy;0(κ͸Gl@l-[+{Yޮڎ0o,VgJGnT|d~.XsÝg#%bj̉Yks)1"ݍaē̟ea'p/p'-@J'kԱm#kqx4UHGC7G`5_wlıq:֐*E*߀8Vx8Dʘ|W#G&:{NA&f338C=2ѷ|&%4x)zW/iuW~{W^]}7BkH)6S[ƛ=CJeG"ׯYB~%4~"_y= :_,;*\IdI0I<{tQ⣗{vԣ';m4)$Ќun=pU1Ev _>jwT GlGG9rD{+l]^"ߛepq`,^{$ b#|\G ň'Fg/w0H-jƢ~]#߯źP;`q6Wml[\Km yGyb[.>5}DsԸ2|v~^k⮪.I\*mp:\vz /.LYa/h`v,ڦ5\b47e^5t-Hd@ &G6UWwNf{;]VTS٢h*пe{O{:|Zw* }nŻө)0qTzDrZH"ICD%͋0r"L&!N@5}"Le0~ }MCd"I(}2R!v P?OAO3 +endstream +endobj + +13 0 obj +6443 +endobj + +14 0 obj +<> +endobj + +15 0 obj +<> +stream +x]n <bL'1&Sp$ ·/\msn])iW,e3T$I*4W sƥ,Iz5;=\2^#Du~ fPƤyb זv?:˟}@SpX5`q\Ѳm+J%E #dI'<ȧs|> +endobj + +17 0 obj +<> +endobj + +18 0 obj +<> +/ProcSet[/PDF/Text/ImageC/ImageI/ImageB] +>> +endobj + +1 0 obj +<>/Contents 2 0 R>> +endobj + +6 0 obj +<> +endobj + +19 0 obj +<> +endobj + +20 0 obj +< +/Producer +/CreationDate(D:20220401110308+02'00')>> +endobj + +xref +0 21 +0000000000 65535 f +0000029273 00000 n +0000000019 00000 n +0000000374 00000 n +0000000394 00000 n +0000011762 00000 n +0000029442 00000 n +0000011784 00000 n +0000020862 00000 n +0000020883 00000 n +0000021078 00000 n +0000021508 00000 n +0000021790 00000 n +0000028320 00000 n +0000028342 00000 n +0000028545 00000 n +0000028904 00000 n +0000029131 00000 n +0000029174 00000 n +0000029541 00000 n +0000029638 00000 n +trailer +< +<77F1D5E7090F17C94EF7E43CF16B016F> ] +/DocChecksum /2F8AB12D558369FCB5C37CF4905E79AE +>> +startxref +29813 +%%EOF diff --git a/src/documents/tests/samples/barcodes/barcode-39-custom.png b/src/documents/tests/samples/barcodes/barcode-39-custom.png new file mode 100644 index 000000000..5c2d7b4f7 Binary files /dev/null and b/src/documents/tests/samples/barcodes/barcode-39-custom.png differ diff --git a/src/documents/tests/samples/barcodes/barcode-qr-custom.pdf b/src/documents/tests/samples/barcodes/barcode-qr-custom.pdf new file mode 100644 index 000000000..0d60b9eed Binary files /dev/null and b/src/documents/tests/samples/barcodes/barcode-qr-custom.pdf differ diff --git a/src/documents/tests/samples/barcodes/barcode-qr-custom.png b/src/documents/tests/samples/barcodes/barcode-qr-custom.png new file mode 100644 index 000000000..6574638bc Binary files /dev/null and b/src/documents/tests/samples/barcodes/barcode-qr-custom.png differ diff --git a/src/documents/tests/samples/patch-code-t-middle.pdf b/src/documents/tests/samples/barcodes/patch-code-t-middle.pdf similarity index 100% rename from src/documents/tests/samples/patch-code-t-middle.pdf rename to src/documents/tests/samples/barcodes/patch-code-t-middle.pdf diff --git a/src/documents/tests/samples/barcodes/patch-code-t-qr.pdf b/src/documents/tests/samples/barcodes/patch-code-t-qr.pdf new file mode 100644 index 000000000..9d2299510 Binary files /dev/null and b/src/documents/tests/samples/barcodes/patch-code-t-qr.pdf differ diff --git a/src/documents/tests/samples/patch-code-t.pbm b/src/documents/tests/samples/barcodes/patch-code-t.pbm similarity index 100% rename from src/documents/tests/samples/patch-code-t.pbm rename to src/documents/tests/samples/barcodes/patch-code-t.pbm diff --git a/src/documents/tests/samples/patch-code-t.pdf b/src/documents/tests/samples/barcodes/patch-code-t.pdf similarity index 100% rename from src/documents/tests/samples/patch-code-t.pdf rename to src/documents/tests/samples/barcodes/patch-code-t.pdf diff --git a/src/documents/tests/samples/barcodes/qr-code-PATCHT.png b/src/documents/tests/samples/barcodes/qr-code-PATCHT.png new file mode 100644 index 000000000..6f1d587ff Binary files /dev/null and b/src/documents/tests/samples/barcodes/qr-code-PATCHT.png differ diff --git a/src/documents/tests/test_tasks.py b/src/documents/tests/test_tasks.py index 2f950e28b..1b59b7109 100644 --- a/src/documents/tests/test_tasks.py +++ b/src/documents/tests/test_tasks.py @@ -1,8 +1,10 @@ import os +import shutil import tempfile from unittest import mock from django.conf import settings +from django.test import override_settings from django.test import TestCase from django.utils import timezone from documents import tasks @@ -95,21 +97,118 @@ class TestTasks(DirectoriesMixin, TestCase): test_file = os.path.join( os.path.dirname(__file__), "samples", - "patch-code-t.pbm", + "barcodes", + "barcode-39-PATCHT.png", ) img = Image.open(test_file) separator_barcode = str(settings.CONSUMER_BARCODE_STRING) self.assertEqual(tasks.barcode_reader(img), [separator_barcode]) def test_barcode_reader2(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "patch-code-t.pbm", + ) + img = Image.open(test_file) + separator_barcode = str(settings.CONSUMER_BARCODE_STRING) + self.assertEqual(tasks.barcode_reader(img), [separator_barcode]) + + def test_barcode_reader_distorsion(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-39-PATCHT-distorsion.png", + ) + img = Image.open(test_file) + separator_barcode = str(settings.CONSUMER_BARCODE_STRING) + self.assertEqual(tasks.barcode_reader(img), [separator_barcode]) + + def test_barcode_reader_distorsion2(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-39-PATCHT-distorsion2.png", + ) + img = Image.open(test_file) + separator_barcode = str(settings.CONSUMER_BARCODE_STRING) + self.assertEqual(tasks.barcode_reader(img), [separator_barcode]) + + def test_barcode_reader_unreadable(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-39-PATCHT-unreadable.png", + ) + img = Image.open(test_file) + self.assertEqual(tasks.barcode_reader(img), []) + + def test_barcode_reader_qr(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "qr-code-PATCHT.png", + ) + img = Image.open(test_file) + separator_barcode = str(settings.CONSUMER_BARCODE_STRING) + self.assertEqual(tasks.barcode_reader(img), [separator_barcode]) + + def test_barcode_reader_128(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-128-PATCHT.png", + ) + img = Image.open(test_file) + separator_barcode = str(settings.CONSUMER_BARCODE_STRING) + self.assertEqual(tasks.barcode_reader(img), [separator_barcode]) + + def test_barcode_reader_no_barcode(self): test_file = os.path.join(os.path.dirname(__file__), "samples", "simple.png") img = Image.open(test_file) self.assertEqual(tasks.barcode_reader(img), []) + def test_barcode_reader_custom_separator(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-39-custom.png", + ) + img = Image.open(test_file) + self.assertEqual(tasks.barcode_reader(img), ["CUSTOM BARCODE"]) + + def test_barcode_reader_custom_qr_separator(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-qr-custom.png", + ) + img = Image.open(test_file) + self.assertEqual(tasks.barcode_reader(img), ["CUSTOM BARCODE"]) + + def test_barcode_reader_custom_128_separator(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-128-custom.png", + ) + img = Image.open(test_file) + self.assertEqual(tasks.barcode_reader(img), ["CUSTOM BARCODE"]) + def test_scan_file_for_separating_barcodes(self): test_file = os.path.join( os.path.dirname(__file__), "samples", + "barcodes", "patch-code-t.pdf", ) pages = tasks.scan_file_for_separating_barcodes(test_file) @@ -124,15 +223,70 @@ class TestTasks(DirectoriesMixin, TestCase): test_file = os.path.join( os.path.dirname(__file__), "samples", + "barcodes", "patch-code-t-middle.pdf", ) pages = tasks.scan_file_for_separating_barcodes(test_file) self.assertEqual(pages, [1]) + def test_scan_file_for_separating_qr_barcodes(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "patch-code-t-qr.pdf", + ) + pages = tasks.scan_file_for_separating_barcodes(test_file) + self.assertEqual(pages, [0]) + + @override_settings(CONSUMER_BARCODE_STRING="CUSTOM BARCODE") + def test_scan_file_for_separating_custom_barcodes(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-39-custom.pdf", + ) + pages = tasks.scan_file_for_separating_barcodes(test_file) + self.assertEqual(pages, [0]) + + @override_settings(CONSUMER_BARCODE_STRING="CUSTOM BARCODE") + def test_scan_file_for_separating_custom_qr_barcodes(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-qr-custom.pdf", + ) + pages = tasks.scan_file_for_separating_barcodes(test_file) + self.assertEqual(pages, [0]) + + @override_settings(CONSUMER_BARCODE_STRING="CUSTOM BARCODE") + def test_scan_file_for_separating_custom_128_barcodes(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-128-custom.pdf", + ) + pages = tasks.scan_file_for_separating_barcodes(test_file) + self.assertEqual(pages, [0]) + + def test_scan_file_for_separating_wrong_qr_barcodes(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "barcode-39-custom.pdf", + ) + pages = tasks.scan_file_for_separating_barcodes(test_file) + self.assertEqual(pages, []) + def test_separate_pages(self): test_file = os.path.join( os.path.dirname(__file__), "samples", + "barcodes", "patch-code-t-middle.pdf", ) pages = tasks.separate_pages(test_file, [1]) @@ -142,6 +296,7 @@ class TestTasks(DirectoriesMixin, TestCase): test_file = os.path.join( os.path.dirname(__file__), "samples", + "barcodes", "patch-code-t.pdf", ) tempdir = tempfile.mkdtemp(prefix="paperless-", dir=settings.SCRATCH_DIR) @@ -153,6 +308,7 @@ class TestTasks(DirectoriesMixin, TestCase): test_file = os.path.join( os.path.dirname(__file__), "samples", + "barcodes", "patch-code-t.pdf", ) nonexistingdir = "/nowhere" @@ -172,6 +328,7 @@ class TestTasks(DirectoriesMixin, TestCase): test_file = os.path.join( os.path.dirname(__file__), "samples", + "barcodes", "patch-code-t.pdf", ) tempdir = tempfile.mkdtemp(prefix="paperless-", dir=settings.SCRATCH_DIR) @@ -183,6 +340,7 @@ class TestTasks(DirectoriesMixin, TestCase): test_file = os.path.join( os.path.dirname(__file__), "samples", + "barcodes", "patch-code-t-middle.pdf", ) tempdir = tempfile.mkdtemp(prefix="paperless-", dir=settings.SCRATCH_DIR) @@ -197,6 +355,19 @@ class TestTasks(DirectoriesMixin, TestCase): self.assertTrue(os.path.isfile(target_file1)) self.assertTrue(os.path.isfile(target_file2)) + @override_settings(CONSUMER_ENABLE_BARCODES=True) + def test_consume_barcode_file(self): + test_file = os.path.join( + os.path.dirname(__file__), + "samples", + "barcodes", + "patch-code-t-middle.pdf", + ) + dst = os.path.join(settings.SCRATCH_DIR, "patch-code-t-middle.pd") + shutil.copy(test_file, dst) + + self.assertEqual(tasks.consume_file(dst), "File successfully split") + @mock.patch("documents.tasks.sanity_checker.check_sanity") def test_sanity_check_success(self, m): m.return_value = SanityCheckMessages()