forked from Github/guess-filename.py
257 lines
16 KiB
Python
257 lines
16 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8; mode: python; -*-
|
|
# Time-stamp: <2016-03-07 13:52:10 vk>
|
|
|
|
import unittest
|
|
import logging
|
|
import tempfile
|
|
import os
|
|
import os.path
|
|
import sys
|
|
from guessfilename import GuessFilename
|
|
|
|
|
|
class TestGuessFilename(unittest.TestCase):
|
|
|
|
guess_filename = None
|
|
|
|
def handle_logging(self, verbose=False, quiet=False):
|
|
"""Log handling and configuration"""
|
|
|
|
if verbose:
|
|
FORMAT = "%(levelname)-8s %(asctime)-15s %(message)s"
|
|
logging.basicConfig(level=logging.DEBUG, format=FORMAT)
|
|
elif quiet:
|
|
FORMAT = "%(levelname)-8s %(message)s"
|
|
logging.basicConfig(level=logging.ERROR, format=FORMAT)
|
|
else:
|
|
FORMAT = "%(levelname)-8s %(message)s"
|
|
logging.basicConfig(level=logging.INFO, format=FORMAT)
|
|
|
|
def setUp(self):
|
|
verbose = True
|
|
quiet = False
|
|
|
|
CONFIGDIR = os.path.join(os.path.expanduser("~"), ".config/guessfilename")
|
|
sys.path.insert(0, CONFIGDIR) # add CONFIGDIR to Python path in order to find config file
|
|
try:
|
|
import guessfilenameconfig
|
|
except ImportError:
|
|
print "Could not find \"guessfilenameconfig.py\" in directory \"" + CONFIGDIR + "\".\nPlease take a look at \"guessfilenameconfig-TEMPLATE.py\", copy it, and configure accordingly."
|
|
sys.exit(1)
|
|
|
|
self.handle_logging(verbose, quiet)
|
|
self.guess_filename = GuessFilename(guessfilenameconfig, logging)
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def test_rename_file(self):
|
|
|
|
tmp_oldfile1 = tempfile.mkstemp()[1]
|
|
oldbasename = os.path.basename(tmp_oldfile1)
|
|
dirname = os.path.abspath(os.path.dirname(tmp_oldfile1))
|
|
newbasename = u'test_rename_file'
|
|
newfilename = os.path.join(dirname, newbasename)
|
|
|
|
self.assertTrue(os.path.isfile(tmp_oldfile1))
|
|
self.assertFalse(os.path.isfile(newfilename))
|
|
|
|
# return False if files are identical
|
|
self.assertFalse(self.guess_filename.rename_file(dirname, oldbasename, oldbasename, dryrun=True, quiet=True))
|
|
|
|
# return False if original file does not exist
|
|
self.assertFalse(self.guess_filename.rename_file(dirname, "test_rename_file_this-is-a-non-existing-file", oldbasename, dryrun=True, quiet=True))
|
|
|
|
# return False if target filename does exist
|
|
tmp_oldfile2 = tempfile.mkstemp()[1]
|
|
self.assertTrue(os.path.isfile(tmp_oldfile2))
|
|
oldbasename2 = os.path.basename(tmp_oldfile2)
|
|
self.assertFalse(self.guess_filename.rename_file(dirname, oldbasename, oldbasename2, dryrun=True, quiet=True))
|
|
os.remove(tmp_oldfile2)
|
|
|
|
# no change with dryrun set:
|
|
self.assertTrue(self.guess_filename.rename_file(dirname, oldbasename, newbasename, dryrun=True, quiet=True))
|
|
self.assertTrue(os.path.isfile(tmp_oldfile1))
|
|
self.assertFalse(os.path.isfile(newfilename))
|
|
|
|
# do rename:
|
|
self.assertTrue(self.guess_filename.rename_file(dirname, oldbasename, newbasename, dryrun=False, quiet=True))
|
|
self.assertFalse(os.path.isfile(tmp_oldfile1))
|
|
self.assertTrue(os.path.isfile(newfilename))
|
|
|
|
os.remove(newfilename)
|
|
|
|
def test_adding_tags(self):
|
|
|
|
self.assertEquals(self.guess_filename.adding_tags(['foo'], ['bar']), ['foo', 'bar'])
|
|
self.assertEquals(self.guess_filename.adding_tags([], ['bar']), ['bar'])
|
|
self.assertEquals(self.guess_filename.adding_tags(['foo'], ['bar', u'baz']), ['foo', 'bar', 'baz'])
|
|
self.assertEquals(self.guess_filename.adding_tags(['foo'], [42]), ['foo', 42])
|
|
|
|
def test_derive_new_filename_from_old_filename(self):
|
|
|
|
self.assertEquals(self.guess_filename.derive_new_filename_from_old_filename(u"2016-03-05 a1 12,34 €.pdf"),
|
|
u"2016-03-05 A1 Festnetz-Internet 12,34€ -- scan finance bill.pdf")
|
|
self.assertEquals(self.guess_filename.derive_new_filename_from_old_filename(u"2016-03-05 A1 12.34 EUR.pdf"),
|
|
u"2016-03-05 A1 Festnetz-Internet 12.34€ -- scan finance bill.pdf")
|
|
|
|
# 2016-01-19--2016-02-12 benutzter GVB 10er Block -- scan transportation graz.pdf
|
|
self.assertEquals(self.guess_filename.derive_new_filename_from_old_filename(u"2016-03-05 10er.pdf"),
|
|
u"2016-03-05 benutzter GVB 10er Block -- scan transportation graz.pdf")
|
|
self.assertEquals(self.guess_filename.derive_new_filename_from_old_filename(u"2016-01-19--2016-02-12 10er GVB.pdf"),
|
|
u"2016-01-19--2016-02-12 benutzter GVB 10er Block -- scan transportation graz.pdf")
|
|
self.assertEquals(self.guess_filename.derive_new_filename_from_old_filename(u"2016-01-19--2016-02-12 10er GVB -- foobar.pdf"),
|
|
u"2016-01-19--2016-02-12 benutzter GVB 10er Block -- foobar scan transportation graz.pdf")
|
|
|
|
def test_contains_one_of(self):
|
|
|
|
self.assertTrue(self.guess_filename.contains_one_of(u"foo bar baz", ['foo']))
|
|
self.assertTrue(self.guess_filename.contains_one_of(u"foo bar baz", [u'foo']))
|
|
self.assertTrue(self.guess_filename.contains_one_of(u"foo bar baz", [u'bar']))
|
|
self.assertTrue(self.guess_filename.contains_one_of(u"foo bar baz", [u'ba']))
|
|
self.assertTrue(self.guess_filename.contains_one_of(u"foo bar baz", [u'x', u'ba', u'yuio']))
|
|
self.assertFalse(self.guess_filename.contains_one_of(u"foo bar baz", ['xfoo']))
|
|
self.assertFalse(self.guess_filename.contains_one_of(u"foo bar baz", [u'xfoo']))
|
|
self.assertFalse(self.guess_filename.contains_one_of(u"foo bar baz", [u'xbar']))
|
|
self.assertFalse(self.guess_filename.contains_one_of(u"foo bar baz", [u'xba']))
|
|
self.assertFalse(self.guess_filename.contains_one_of(u"foo bar baz", [u'x', u'xba', u'yuio']))
|
|
|
|
def test_fuzzy_contains_all_of(self):
|
|
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", ['foo']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", [u'foo']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", [u'bar']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", [u'ba']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", [u'foo', u"bar", u"baz"]))
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", [u'x', u'ba', u'yuio']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", ['xfoo']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", [u'xfoo']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", [u'xbar']))
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", [u'xba', u"12345"]))
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_all_of(u"foo bar baz", [u'x', u'xba', u'yuio']))
|
|
|
|
def test_fuzzy_contains_one_of(self):
|
|
|
|
# comparing exact strings:
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", ['foo']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'foo']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'bar']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'ba']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'x', u'ba', u'yuio']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u'12345']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u'12345']))
|
|
|
|
# fuzzy similarities:
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", ['xfoo']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'xfoo']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'xbar']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'xba']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'x', u'xba', u'yuio']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u'7234567880']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u'Rundemummer 7234567880']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u'Rumdemummer 7234567880']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u'Rundemummer 1234581388']))
|
|
self.assertTrue(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u'Rumdemummer 1234581388']))
|
|
|
|
# fuzzy non-matches:
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u' 345 ']))
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u'1234581388']))
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'xyz']))
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'111']))
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'xby']))
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_one_of(u"foo bar baz", [u'x', u'yyy', u'yuio']))
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u'0987654321']))
|
|
self.assertFalse(self.guess_filename.fuzzy_contains_one_of(u"Kundennummer 1234567890", [u'Rumdemummer 1234555555']))
|
|
|
|
def test_has_euro_charge(self):
|
|
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"12,34EUR"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"12.34EUR"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"12,34 EUR"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"12.34 EUR"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"foo bar 12,34 EUR baz"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"foo bar 12.34 EUR baz"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"foo bar 12 EUR baz"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"foo bar 12EUR baz"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"foo bar 12.34 EUR baz.extension"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"12,34€"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"12.34€"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"12,34 €"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"12.34 €"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"foo bar 12,34 € baz"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"foo bar 12.34 € baz"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"foo bar 12.34 € baz.extension"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"2016-03-05 A1 12.34 EUR.pdf"))
|
|
self.assertTrue(self.guess_filename.has_euro_charge(u"2016-03-05 A1 Festnetz-Internet 12.34 € -- scan finance bill.pdf"))
|
|
self.assertFalse(self.guess_filename.has_euro_charge(u"1234"))
|
|
self.assertFalse(self.guess_filename.has_euro_charge(u"foo bar baz"))
|
|
self.assertFalse(self.guess_filename.has_euro_charge(u"1234eur"))
|
|
self.assertFalse(self.guess_filename.has_euro_charge(u"foo 12 bar"))
|
|
self.assertFalse(self.guess_filename.has_euro_charge(u"foo 1234 bar"))
|
|
self.assertFalse(self.guess_filename.has_euro_charge(u"foo 12,34 bar"))
|
|
self.assertFalse(self.guess_filename.has_euro_charge(u"foo 12.34 bar"))
|
|
|
|
def test_get_euro_charge(self):
|
|
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"12,34EUR"), "12,34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"12.34EUR"), "12.34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"12,34 EUR"), "12,34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"12.34 EUR"), "12.34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"foo bar 12,34 EUR baz"), "12,34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"foo bar 12.34 EUR baz"), "12.34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"foo bar 12 EUR baz"), "12")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"foo bar 12EUR baz"), "12")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"foo bar 12.34 EUR baz.extension"), "12.34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"12,34€"), "12,34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"12.34€"), "12.34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"12,34 €"), "12,34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"12.34 €"), "12.34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"foo bar 12,34 € baz"), "12,34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"foo bar 12.34 € baz"), "12.34")
|
|
self.assertEquals(self.guess_filename.get_euro_charge(u"foo bar 12.34 € baz.extension"), "12.34")
|
|
self.assertFalse(self.guess_filename.get_euro_charge(u"1234"))
|
|
self.assertFalse(self.guess_filename.get_euro_charge(u"foo bar baz"))
|
|
self.assertFalse(self.guess_filename.get_euro_charge(u"1234eur"))
|
|
self.assertFalse(self.guess_filename.get_euro_charge(u"foo 12 bar"))
|
|
self.assertFalse(self.guess_filename.get_euro_charge(u"foo 1234 bar"))
|
|
self.assertFalse(self.guess_filename.get_euro_charge(u"foo 12,34 bar"))
|
|
self.assertFalse(self.guess_filename.get_euro_charge(u"foo 12.34 bar"))
|
|
|
|
def test_split_filename_entities(self):
|
|
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"2016-03-05T23.59.42--2017-04-06T12.13.14 foo bar -- eins zwei.extension"),
|
|
(u"2016-03-05T23.59.42--2017-04-06T12.13.14", u"foo bar", [u"eins", u"zwei"], u"extension"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"2016-03-05T23.59.42-2017-04-06T12.13.14 foo bar -- eins zwei.extension"),
|
|
(u"2016-03-05T23.59.42-2017-04-06T12.13.14", u"foo bar", [u"eins", u"zwei"], u"extension"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"2016-03-05T23.59--2017-04-06T12.13 foo - bar.baz - zum -- eins zwei.extension"),
|
|
(u"2016-03-05T23.59--2017-04-06T12.13", u"foo - bar.baz - zum", [u"eins", u"zwei"], u"extension"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"2016-03-05T23.59.42 foo - bar.baz - zum -- eins zwei.extension"),
|
|
(u"2016-03-05T23.59.42", u"foo - bar.baz - zum", [u"eins", u"zwei"], u"extension"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"2016-03-05T23.59.42 foo bar -- eins zwei.extension"),
|
|
(u"2016-03-05T23.59.42", u"foo bar", [u"eins", u"zwei"], u"extension"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"2016-03-05T23.59--2017-04-06T12.13 foo bar -- eins zwei.extension"),
|
|
(u"2016-03-05T23.59--2017-04-06T12.13", u"foo bar", [u"eins", u"zwei"], u"extension"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"2016-03-05T23.59--2017-04-06T12.13 foo bar.extension"),
|
|
(u"2016-03-05T23.59--2017-04-06T12.13", u"foo bar", [], u"extension"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"2016-03-05T23.59 foo bar.extension"),
|
|
(u"2016-03-05T23.59", u"foo bar", [], u"extension"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"foo bar.extension"),
|
|
(None, u"foo bar", [], u"extension"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"foo bar"),
|
|
(None, u"foo bar", [], None))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"foo.bar"),
|
|
(None, u"foo", [], u"bar"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"foo -- bar"),
|
|
(None, u"foo", [u"bar"], None))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"foo -- bar.baz"),
|
|
(None, u"foo", [u"bar"], u"baz"))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u" -- "),
|
|
(None, u' -- ', [], None))
|
|
self.assertEqual(self.guess_filename.split_filename_entities(u"."),
|
|
(None, u'.', [], None))
|
|
|
|
# Local Variables:
|
|
# mode: flyspell
|
|
# eval: (ispell-change-dictionary "en_US")
|
|
# End:
|