"""Format details. """ import sys import io import os from datetime import datetime from nose.tools import * import rarfile def render_date(dt): if isinstance(dt, datetime): return dt.isoformat('T') elif isinstance(dt, tuple): return '%04d-%02d-%02d %02d:%02d:%02d' % dt else: return dt def mkitem(**kwargs): res = {} for k in kwargs: if kwargs[k] is not None: res[k] = kwargs[k] return res def dumparc(rf): res = [] for item in rf.infolist(): info = mkitem(fn=item.filename, file_size=item.file_size, compress_size=item.compress_size, CRC=item.CRC, date_time=render_date(item.date_time), arctime=render_date(item.arctime), mtime=render_date(item.mtime), atime=render_date(item.atime), ctime=render_date(item.ctime), comment=item.comment, extract_version=item.extract_version, compress_type=item.compress_type, mode=item.mode, host_os=item.host_os) res.append(info) return res def diffs(a, b): if len(a) != len(b): return 'Different lengths' problems = [] for i, xa in enumerate(a): xb = b[i] for k in xa: if k not in xb: problems.append('NewKey(%d,%s)=%r' % (i, k, xa[k])) for k in xb: if k not in xa: problems.append('MissingKey(%d,%s)=%r' % (i, k, xb[k])) for k in xa: if k in xb and xa[k] != xb[k]: problems.append('ErrValue(%d,%s):got=%r/exp=%r' % (i, k, xa[k], xb[k])) return '; '.join(problems) def cmp_struct(a, b): eq_(a, b, diffs(a, b)) # # test start # def test_rar3_header_encryption(): r = rarfile.RarFile('test/files/rar3-comment-hpsw.rar', 'r') eq_(r.needs_password(), True) eq_(r.comment, None) eq_(r.namelist(), []) try: r.setpassword('password') assert_true(r.needs_password()) eq_(r.namelist(), [u'file1.txt', u'file2.txt']) assert_not_equal(r.comment, None) eq_(r.comment, 'RARcomment\n') except rarfile.NoCrypto: pass def test_rar5_header_encryption(): r = rarfile.RarFile('test/files/rar5-hpsw.rar') eq_(r.needs_password(), True) eq_(r.comment, None) eq_(r.namelist(), []) try: r.setpassword('password') assert_true(r.needs_password()) eq_(r.namelist(), [u'stest1.txt', u'stest2.txt']) assert_not_equal(r.comment, None) eq_(r.comment, 'RAR5 archive - hdr-password\n') except rarfile.NoCrypto: pass r.close() def get_vol_info(extver=20, tz='', hr='11'): return [ mkitem(CRC=1352324940, date_time='2016-05-24 %s:42:37%s' % (hr, ''), mtime='2016-05-24T%s:42:37%s' % (hr, tz), compress_type=48, compress_size=205000, extract_version=extver, file_size=205000, mode=33204, host_os=3, fn=u'vols/bigfile.txt'), mkitem(CRC=3498712966, date_time='2016-05-24 %s:42:43%s' % (hr, ''), mtime='2016-05-24T%s:42:43%s' % (hr, tz), extract_version=extver, compress_type=48, compress_size=2050, file_size=2050, mode=33204, host_os=3, fn=u'vols/smallfile.txt')] def test_rar3_vols(): r = rarfile.RarFile('test/files/rar3-vols.part1.rar') eq_(r.needs_password(), False) eq_(r.comment, None) eq_(r.strerror(), None) cmp_struct(dumparc(r), get_vol_info()) eq_(r.volumelist(), [ 'test/files/rar3-vols.part1.rar', 'test/files/rar3-vols.part2.rar', 'test/files/rar3-vols.part3.rar']) def test_rar3_oldvols(): r = rarfile.RarFile('test/files/rar3-old.rar') eq_(r.needs_password(), False) eq_(r.comment, None) eq_(r.strerror(), None) cmp_struct(dumparc(r), get_vol_info()) eq_(r.volumelist(), [ 'test/files/rar3-old.rar', 'test/files/rar3-old.r00', 'test/files/rar3-old.r01']) def test_rar5_vols(): r = rarfile.RarFile('test/files/rar5-vols.part1.rar') eq_(r.needs_password(), False) eq_(r.comment, None) eq_(r.strerror(), None) cmp_struct(dumparc(r), get_vol_info(50, '+00:00', '08')) eq_(r.volumelist(), [ 'test/files/rar5-vols.part1.rar', 'test/files/rar5-vols.part2.rar', 'test/files/rar5-vols.part3.rar']) def expect_ctime(mtime, ctime): return [mkitem( mtime=mtime, date_time=mtime.split('.')[0].replace('T', ' '), ctime=ctime, compress_size=0, file_size=0, CRC=0, fn=u'afile.txt', extract_version=29, compress_type=48, mode=32, host_os=2)] def test_rar3_ctime0(): r = rarfile.RarFile('test/files/ctime0.rar') cmp_struct(dumparc(r), expect_ctime('2011-05-10T21:28:47.899345', None)) def test_rar3_ctime1(): r = rarfile.RarFile('test/files/ctime1.rar') cmp_struct(dumparc(r), expect_ctime('2011-05-10T21:28:47.899345', '2011-05-10T21:28:47')) def test_rar3_ctime2(): r = rarfile.RarFile('test/files/ctime2.rar') cmp_struct(dumparc(r), expect_ctime('2011-05-10T21:28:47.899345', '2011-05-10T21:28:47.897843')) def test_rar3_ctime3(): r = rarfile.RarFile('test/files/ctime3.rar') cmp_struct(dumparc(r), expect_ctime('2011-05-10T21:28:47.899345', '2011-05-10T21:28:47.899328')) def test_rar3_ctime4(): r = rarfile.RarFile('test/files/ctime4.rar') cmp_struct(dumparc(r), expect_ctime('2011-05-10T21:28:47.899345', '2011-05-10T21:28:47.899345')) def test_rar5_times(): r = rarfile.RarFile('test/files/rar5-times.rar') cmp_struct(dumparc(r), [mkitem( fn=u'stest1.txt', file_size=2048, compress_size=55, compress_type=rarfile.RAR_M3, extract_version=50, host_os=rarfile.RAR_OS_UNIX, mode=33188, date_time='2011-06-12 09:53:33', mtime='2011-06-12T09:53:33+00:00', atime='2016-05-22T09:12:36+00:00', CRC=3317163682 )]) def test_oldvols(): eq_(rarfile._next_oldvol('qq00.part0.rar'), 'qq00.part0.r00') eq_(rarfile._next_oldvol('qq00.part0.r00'), 'qq00.part0.r01') eq_(rarfile._next_oldvol('qq00.part0.r29'), 'qq00.part0.r30') eq_(rarfile._next_oldvol('qq00.part0.r99'), 'qq00.part0.s00') def test_newvols(): eq_(rarfile._next_newvol('qq00.part0.rar'), 'qq00.part1.rar') eq_(rarfile._next_newvol('qq00.part09.rar'), 'qq00.part10.rar') eq_(rarfile._next_newvol('qq00.part99.rar'), 'qq00.paru00.rar') @raises(rarfile.BadRarName) def test_newvols_err(): rarfile._next_newvol('xx.rar')