-
Notifications
You must be signed in to change notification settings - Fork 33
/
filewatcher.py
165 lines (132 loc) · 5 KB
/
filewatcher.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (c) 2010.
# SMHI,
# Folkborgsvägen 1,
# Norrköping,
# Sweden
# Author(s):
# Martin Raspaud <martin.raspaud@smhi.se>
# This file is part of mpop.
# mpop 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 3 of the License, or (at your option) any later
# version.
# mpop 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
# mpop. If not, see <http://www.gnu.org/licenses/>.
"""Watch files coming in a given directory.
"""
import datetime
import glob
import os
from Queue import Empty
from threading import Thread, Condition
import time
from mpop.saturn import LOG
ZERO = datetime.timedelta(seconds=0)
class FileWatcher(Thread):
"""Looks for new files, and queues them.
"""
def __init__(self, filename_template, file_queue, frequency):
"""Looks for new files arriving at the given *frequency*, and queues
them.
"""
Thread.__init__(self)
self.queue = file_queue
self.template = filename_template
self.frequency = datetime.timedelta(minutes=frequency)
self.running = True
self.cond = Condition()
def terminate(self):
"""Terminate thread.
"""
self.running = False
self.cond.acquire()
self.cond.notify()
self.cond.release()
LOG.debug("Termination request received in FileWatcher")
def wait(self, secs):
if self.running:
self.cond.wait(secs)
def run(self):
"""Run the file watcher.
"""
filelist = set()
while self.running:
self.cond.acquire()
if isinstance(self.template, (list, tuple)):
new_filelist = []
for template in self.template:
new_filelist += glob.glob(template)
new_filelist = set(new_filelist)
else:
new_filelist = set(glob.glob(self.template))
files_to_process = list(new_filelist - filelist)
filelist = new_filelist
files_dict = {}
for fil in files_to_process:
files_dict[fil] = os.path.getmtime(fil)
files_to_process.sort(lambda x, y: cmp(files_dict[x],
files_dict[y]))
if len(files_to_process) != 0 and self.running:
sleep_time = 8
times = []
for i in files_to_process:
LOG.debug("queueing %s..."%i)
self.queue.put(i)
times.append(os.stat(i).st_ctime)
times.sort()
since_creation = datetime.timedelta(seconds=time.time() -
times[-1])
if(self.frequency > since_creation):
to_wait = self.frequency - since_creation
LOG.info("Waiting at least "+str(to_wait)+" for next file")
sleep_time = (to_wait.seconds +
to_wait.microseconds / 1000000.0)
self.wait(sleep_time)
elif self.running:
LOG.info("no new file has come, waiting %s secs"
%str(sleep_time))
self.wait(sleep_time)
if sleep_time < 60:
sleep_time *= 2
self.cond.release()
LOG.info("FileWatcher terminated.")
class FileProcessor(Thread):
"""Execute *fun* on filenames provided by from *file_queue*. If *refresh*
is a positive number, run *fun* every given number of seconds with None as
argument.
"""
def __init__(self, file_queue, fun, refresh=None):
Thread.__init__(self)
self.queue = file_queue
self.fun = fun
self.running = True
self.refresh = refresh
def terminate(self):
"""Terminate thread.
"""
self.running = False
LOG.debug("Termination request received in FileProcessor")
def run(self):
"""Execute the given function on files from the file queue.
"""
while self.running:
try:
filename = self.queue.get(block=True, timeout=self.refresh)
LOG.debug("processing %s"%filename)
except Empty:
filename = None
try:
self.fun(filename)
except:
LOG.exception("Something wrong happened in %s for %s. Skipping."
%(str(self.fun), filename))
LOG.info("FileProcessor terminated.")
def stop(self):
"""Stops a running process.
"""
self.running = False