X-Git-Url: https://vcs.maemo.org/git/?a=blobdiff_plain;f=dbuscron%2Fparser.py;h=73ab3a04fc83ecf24a186086a7a48975757baacc;hb=3199277f113e3c9238f3d4d48b03df7e3ca43d55;hp=f31d3ee2e0f66344f071f5d393c2bfae64613071;hpb=483c6ab91a57a80bd0ff0a1703e90d2a76ff62e1;p=dbuscron

diff --git a/dbuscron/parser.py b/dbuscron/parser.py
index f31d3ee..73ab3a0 100644
--- a/dbuscron/parser.py
+++ b/dbuscron/parser.py
@@ -44,7 +44,25 @@ class CrontabParserError(SyntaxError):
 
         SyntaxError.__init__(self, msg)
 
-class CrontabParser(object):
+def CrontabParser(*filenames):
+    for filename in filenames:
+        if not os.path.exists(filename):
+            continue
+
+        if os.path.isfile(filename):
+            parser_class = FileParser
+
+        elif os.path.isdir(filename):
+            parser_class = DirectoryParser
+
+        else:
+            raise SystemError("I can parse only directories or simple files.")
+
+        return parser_class(filename)
+
+    raise SystemError("Can't find any config file or directory.")
+
+class FileParser(object):
     __fields_sep = re.compile(r'\s+')
     __envvar_sep = re.compile(r'\s*=\s*')
     __fields_chk = {
@@ -69,13 +87,17 @@ class CrontabParser(object):
             ]
 
     def __init__(self, fname):
-        self.__bus = DbusBus()
-        self.__filename = fname
-        self.__environ = dict()
+        self._bus = DbusBus()
+        self._filename = fname
+        self._environ = dict()
 
     @property
     def environ(self):
-        return self.__environ
+        return self._environ
+
+    @property
+    def filename(self):
+        return self._filename
 
     def _iterate_file(self, filename):
         # bus type sender interface path member destination args command
@@ -108,9 +130,9 @@ class CrontabParser(object):
                 for r in product(*rule):
                     r = list(r)
                     if r[0] == 'S':
-                        r[0] = self.__bus.system
+                        r[0] = self._bus.system
                     elif r[0] == 's':
-                        r[0] = self.__bus.session
+                        r[0] = self._bus.session
                     else:
                         raise CrontabParserError('Unexpected bus value', lineno, expected=('S', 's', '*'))
 
@@ -132,36 +154,33 @@ class CrontabParser(object):
                     yield ruled, command
 
     def __iter__(self):
-        return self._iterate_file(self.__filename)
+        return self._iterate_file(self._filename)
 
-class DirectoryParser(CrontabParser):
+class DirectoryParser(FileParser):
 
     def __init__(self, dirname, recursive=False):
-        self.__dirname = dirname
         self.__recursive = recursive
-        super(DirectoryParser, self).__init__(None)
+        super(DirectoryParser, self).__init__(dirname)
 
     def _dirwalker_plain(self):
-        for i in os.listdir(self.__dirname):
-            if os.path.isfile(i):
-                yield i
+        for i in os.listdir(self._filename):
+            f = os.path.join(self._filename, i)
+            if os.path.isfile(f):
+                yield f
 
     def _dirwalker_recursive(self):
-        for r, d, f in os.walk(self.__dirname):
+        for r, d, f in os.walk(self._filename):
             for i in f:
-                yield i
+                yield os.path.join(r, i)
 
     def __iter__(self):
-
         if self.__recursive:
             dirwalker = self._dirwalker_recursive
         else:
             dirwalker = self._dirwalker_plain
 
         for fname in dirwalker():
-            fullname = os.path.join(self.__dirname, fname)
-            self.__filename = fullname
-            for item in self._iterate_file(fullname):
+            for item in self._iterate_file(fname):
                 yield item
 
 def OptionsParser(args=None, help=u'', **opts):