From: allfro <ndouba@gmail.com>
Date: Mon, 8 Apr 2013 01:21:41 +0000 (-0400)
Subject: Added `image_resources()` function to get all images from transform pack
X-Git-Url: https://git.nothing2do.fr/?a=commitdiff_plain;h=5aafbb5eef06305f8a8fd489c55b5ff71b16580e;p=get-hack-src.git

Added `image_resources()` function to get all images from transform pack
---

diff --git a/src/canari/resource.py b/src/canari/resource.py
index 3a928a2..c8e8b6f 100644
--- a/src/canari/resource.py
+++ b/src/canari/resource.py
@@ -2,7 +2,8 @@
 
 from utils.stack import modulecallee
 
-from pkg_resources import resource_filename
+from imghdr import what
+from pkg_resources import resource_filename, resource_listdir, resource_isdir
 
 __author__ = 'Nadeem Douba'
 __copyright__ = 'Copyright 2012, Canari Project'
@@ -47,5 +48,18 @@ def icon_resource(name, pkg=None):
     return imageicon(pkg, name)
 
 
+def image_resources(pkg=None, dir='resources'):
+    if pkg is None:
+        pkg = modulecallee().__name__.split('.')[0]
+    pkg_dir = '%s.%s' % (pkg, dir)
+    images = []
+    for i in resource_listdir(pkg, dir):
+        fname = resource_filename(pkg_dir, i)
+        if resource_isdir(pkg_dir, i):
+            images.extend(image_resources(pkg_dir, i))
+        elif what(fname) is not None:
+            images.append(fname)
+    return images
+
 # etc
 conf = resource_filename(etc, 'canari.conf')