1
0
mirror of https://github.com/gryf/pygtktalog.git synced 2025-12-19 12:28:13 +01:00

* Added shell script for preparing distfile.

* Added unitests.
 * Added mocks for video files.
 * Added midentify module.
This commit is contained in:
2008-12-15 20:46:26 +00:00
parent 0adcdaba8d
commit 292d290723
13 changed files with 286 additions and 0 deletions

32
prepare_dist_package.sh Executable file
View File

@@ -0,0 +1,32 @@
#!/bin/sh
# remove ~, pyc, pyo files from current directory
mkdir t 2>/dev/null
if [ $? != 0 ]; then
echo "cannot create directory 't': File exist."
echo "Rename it, move or rename, bcoz it's on the way."
exit
fi
cd t
alias ls=ls
PREV=`ls -1 ..|grep bz2|tail -n 1|cut -f '2' -d '_'|cut -f 1 -d '.'`
REV=`svn export svn://10.0.0.10/repos/Python/pyGTKtalog pyGTKtalog |tail -n 1|cut -f 3 -d " "|cut -f 1 -d '.'`
cd pyGTKtalog
find . -name \*~ -exec rm '{}' ';'
find . -name \*pyc -exec rm '{}' ';'
find . -name \*pyo -exec rm '{}' ';'
find . -type d -name .svn -exec rm -fr '{}' ';'
rm -fr db img
rm -fr prepare_dist_package.sh
svn log -r ${PREV}:HEAD -v svn://10.0.0.10/repos/Python/pyGTKtalog > CHANGELOG
cd ..
tar jcf ../pygtktalog_${REV}.tar.bz2 pyGTKtalog
cd ..
rm -fr t

63
src/lib/midentify.py Normal file
View File

@@ -0,0 +1,63 @@
# This Python file uses the following encoding: utf-8
from os import popen
from sys import argv, exit
class Midentify(object):
"""Class for retrive midentify script output and put it in dict.
Usually there is no need for such a detailed movie/clip information.
Midentify script belongs to mplayer package.
"""
def __init__(self, filename):
"""Init class instance. Filename of a video file is required."""
self.filename = filename
self.tags = {}
def get_data(self):
"""return dict with clip information"""
output = popen("midentify \"%s\"" % self.filename).readlines()
for line in output:
line = line.strip()
if "ID_VIDEO_WIDTH" in line:
self.tags['width'] = line.replace("ID_VIDEO_WIDTH=", "")
elif "ID_VIDEO_HEIGHT" in line:
self.tags['height'] = line.replace("ID_VIDEO_HEIGHT=", "")
elif "ID_LENGTH" in line:
length = line.replace("ID_LENGTH=", "")
if "." in length:
length = length.split(".")[0]
seconds = int(length)
if seconds > 0:
hours = seconds / 3600
seconds -= hours * 3600
minutes = seconds / 60
seconds -= minutes * 60
self.tags['length'] = length
length_str = "%02d:%02d:%02d" % (hours, minutes, seconds)
self.tags['duration'] = length_str
elif "ID_DEMUXER" in line:
self.tags['container'] = line.replace("ID_DEMUXER=", "")
elif "ID_VIDEO_FORMAT" in line:
self.tags['video_format'] = line.replace("ID_VIDEO_FORMAT=", "")
elif "ID_VIDEO_CODEC" in line:
self.tags['video_codec'] = line.replace("ID_VIDEO_CODEC=", "")
elif "ID_AUDIO_CODEC" in line:
self.tags['audio_codec'] = line.replace("ID_AUDIO_CODEC=", "")
elif "ID_AUDIO_FORMAT" in line:
self.tags['audio_format'] = line.replace("ID_AUDIO_FORMAT=", "")
elif "ID_AUDIO_NCH" in line:
self.tags['audio_no_channels'] = line.replace("ID_AUDIO_NCH=",
"")
return self.tags
if __name__ == "__main__":
"""run as standalone script"""
if len(argv) < 2:
print "usage: %s filename" % argv[0]
exit()
for arg in argv[1:]:
mid = Midentify(arg)
print mid.get_data()

0
src/test/__init__.py Normal file
View File

BIN
src/test/mocks/m.avi Normal file

Binary file not shown.

BIN
src/test/mocks/m.mkv Normal file

Binary file not shown.

BIN
src/test/mocks/m.mpg Normal file

Binary file not shown.

BIN
src/test/mocks/m.ogm Normal file

Binary file not shown.

BIN
src/test/mocks/m1.avi Normal file

Binary file not shown.

76
src/test/run_tests.py Executable file
View File

@@ -0,0 +1,76 @@
#!/usr/bin/env python
# This Python file uses the following encoding: utf-8
#
# Author: Roman 'gryf' Dobosz gryf@elysium.pl
#
# Copyright (C) 2007 by Roman 'gryf' Dobosz
#
# This file is part of pyGTKtalog.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
# -------------------------------------------------------------------------
import sys
import unittest
from os import path, chdir
import glob
def my_import(module, name):
"""import replacement"""
mod = __import__(module, {}, {}, [name])
components = name.split('.')
for comp in components[1:]:
mod = getattr(mod, comp)
return mod
def setup_path():
"""Sets up the python include paths to include needed directories"""
this_path = path.abspath(path.dirname(__file__))
sys.path = [path.join(this_path, "../../src")] + sys.path
sys.path = [path.join(this_path, "../../src/test")] + sys.path
sys.path = [path.join(this_path, "../../src/test/unit")] + sys.path
return
def build_suite():
"""build suite test from files in unit directory"""
modules = []
classes = []
for fname in glob.glob1('unit', '*_test.py'):
class_name = fname[:-8]
if "_" in class_name:
splited = class_name.split("_")
class_name = 'Test'
for word in splited:
class_name += word.capitalize()
else:
class_name = "Test" + class_name.capitalize()
modules.append(fname[:-3])
classes.append(class_name)
modules = map(__import__, modules)
#my_import(fname[:-3], class_name)
load = unittest.defaultTestLoader.loadTestsFromModule
return unittest.TestSuite(map(load, modules))
#exec('suite.addTest(TL().loadTestsFromTestCase('+class_name+'))')
#return suite
if __name__ == "__main__":
chdir(path.abspath(path.curdir))
setup_path()
unittest.main(defaultTest="build_suite")
#unittest.TextTestRunner().run(build_suite())

View File

View File

@@ -0,0 +1,11 @@
# This Python file uses the following encoding: utf-8
import unittest
class TestDummy(unittest.TestCase):
"""Fake test class"""
def test_dummyMethod(self):
"""Test simple assertion"""
self.assertTrue(True)
if __name__ == "__main__":
unittest.main()

View File

@@ -0,0 +1,11 @@
# This Python file uses the following encoding: utf-8
import unittest
class TestFooBar(unittest.TestCase):
"""Fake test class"""
def test_dummyMethod(self):
"""Test simple assertion"""
self.assertTrue(True)
if __name__ == "__main__":
unittest.main()

View File

@@ -0,0 +1,93 @@
# This Python file uses the following encoding: utf-8
import unittest
from lib.midentify import Midentify
class TestMidentify(unittest.TestCase):
"""Class for retrive midentify script output and put it in dict.
Usually there is no need for such a detailed movie/clip information.
Midentify script belongs to mplayer package.
"""
def test_testAvi(self):
"""test mock avi file, should return dict with expected values"""
avi = Midentify("mocks/m.avi")
result_dict = avi.get_data()
self.assertTrue(len(result_dict) != 0, "result should have lenght > 0")
self.assertEqual(result_dict['audio_format'], '85')
self.assertEqual(result_dict['width'], '128')
self.assertEqual(result_dict['audio_no_channels'], '2')
self.assertEqual(result_dict['height'], '96')
self.assertEqual(result_dict['video_format'], 'XVID')
self.assertEqual(result_dict['length'], '4')
self.assertEqual(result_dict['audio_codec'], 'mp3')
self.assertEqual(result_dict['video_codec'], 'ffodivx')
self.assertEqual(result_dict['duration'], '00:00:04')
self.assertEqual(result_dict['container'], 'avi')
def test_testAvi2(self):
"""test another mock avi file, should return dict with expected
values"""
avi = Midentify("mocks/m1.avi")
result_dict = avi.get_data()
self.assertTrue(len(result_dict) != 0, "result should have lenght > 0")
self.assertEqual(result_dict['audio_format'], '85')
self.assertEqual(result_dict['width'], '128')
self.assertEqual(result_dict['audio_no_channels'], '2')
self.assertEqual(result_dict['height'], '96')
self.assertEqual(result_dict['video_format'], 'H264')
self.assertEqual(result_dict['length'], '4')
self.assertEqual(result_dict['audio_codec'], 'mp3')
self.assertEqual(result_dict['video_codec'], 'ffh264')
self.assertEqual(result_dict['duration'], '00:00:04')
self.assertEqual(result_dict['container'], 'avi')
def test_testMkv(self):
"""test mock mkv file, should return dict with expected values"""
avi = Midentify("mocks/m.mkv")
result_dict = avi.get_data()
self.assertTrue(len(result_dict) != 0, "result should have lenght > 0")
self.assertEqual(result_dict['audio_format'], '8192')
self.assertEqual(result_dict['width'], '128')
self.assertEqual(result_dict['audio_no_channels'], '2')
self.assertEqual(result_dict['height'], '96')
self.assertEqual(result_dict['video_format'], 'mp4v')
self.assertEqual(result_dict['length'], '4')
self.assertEqual(result_dict['audio_codec'], 'a52')
self.assertEqual(result_dict['video_codec'], 'ffodivx')
self.assertEqual(result_dict['duration'], '00:00:04')
self.assertEqual(result_dict['container'], 'mkv')
def test_testMpg(self):
"""test mock mpg file, should return dict with expected values"""
avi = Midentify("mocks/m.mpg")
result_dict = avi.get_data()
self.assertTrue(len(result_dict) != 0, "result should have lenght > 0")
self.assertFalse(result_dict.has_key('audio_format'))
self.assertEqual(result_dict['width'], '128')
self.assertFalse(result_dict.has_key('audio_no_channels'))
self.assertEqual(result_dict['height'], '96')
self.assertEqual(result_dict['video_format'], '0x10000001')
self.assertFalse(result_dict.has_key('lenght'))
self.assertFalse(result_dict.has_key('audio_codec'))
self.assertEqual(result_dict['video_codec'], 'mpegpes')
self.assertFalse(result_dict.has_key('duration'))
self.assertEqual(result_dict['container'], 'mpeges')
def test_testOgm(self):
"""test mock ogm file, should return dict with expected values"""
avi = Midentify("mocks/m.ogm")
result_dict = avi.get_data()
self.assertTrue(len(result_dict) != 0, "result should have lenght > 0")
self.assertEqual(result_dict['audio_format'], '8192')
self.assertEqual(result_dict['width'], '160')
self.assertEqual(result_dict['audio_no_channels'], '2')
self.assertEqual(result_dict['height'], '120')
self.assertEqual(result_dict['video_format'], 'H264')
self.assertEqual(result_dict['length'], '4')
self.assertEqual(result_dict['audio_codec'], 'a52')
self.assertEqual(result_dict['video_codec'], 'ffh264')
self.assertEqual(result_dict['duration'], '00:00:04')
self.assertEqual(result_dict['container'], 'ogg')
if __name__ == "__main__":
unittest.main()