137 lines
4.6 KiB
Python
137 lines
4.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright (C) 2007-2014 Clement Lefebvre <root@linuxmint.com>
|
|
# Copyright (C) 2015 Martin Wimpress <code@ubuntu-mate.org>
|
|
#
|
|
# 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 os
|
|
import os.path
|
|
import threading
|
|
import time
|
|
from gi.repository import GLib
|
|
|
|
try:
|
|
import pyinotify
|
|
hasInotify = True
|
|
except ImportError:
|
|
hasInotify = False
|
|
|
|
if hasInotify:
|
|
class FileMonitor(object):
|
|
def __init__( self ):
|
|
self.monitorId = 0
|
|
self.wm = pyinotify.WatchManager()
|
|
self.wdds = {}
|
|
self.callbacks = {}
|
|
self.notifier = pyinotify.ThreadedNotifier(self.wm, self.fileChanged)
|
|
self.notifier.setDaemon( True )
|
|
self.notifier.start()
|
|
|
|
|
|
def addMonitor( self, filename, callback, args = None ):
|
|
try:
|
|
mask = pyinotify.IN_DELETE | pyinotify.IN_CREATE | pyinotify.IN_MODIFY
|
|
mId = self.wm.add_watch( filename, mask, rec = True)[filename]
|
|
if mId >= 0:
|
|
self.callbacks[mId] = ( callback, args )
|
|
except Exception as detail:
|
|
mId = 0
|
|
return mId
|
|
|
|
def removeMonitor( self, monitorId ):
|
|
if monitorId in self.callbacks:
|
|
self.wm.rm_watch( monitorId )
|
|
del self.callbacks[monitorId]
|
|
|
|
def fileChanged(self, event ):
|
|
if event.wd in self.callbacks:
|
|
callback = self.callbacks[event.wd]
|
|
if callback[1]:
|
|
GLib.idle_add( callback[0], callback[1] )
|
|
else:
|
|
GLib.idle_add( callback[0] )
|
|
else:
|
|
|
|
class _MonitoredFile( object ):
|
|
def __init__( self, filename, callback, monitorId, args ):
|
|
self.filename = filename
|
|
self.callback = callback
|
|
self.monitorId = monitorId
|
|
self.args = args
|
|
self.exists = os.path.exists( self.filename )
|
|
if self.exists:
|
|
self.mtime = os.stat( filename ).st_mtime
|
|
else:
|
|
self.mtime = 0
|
|
|
|
def hasChanged( self ):
|
|
if os.path.exists( self.filename ):
|
|
if not self.exists:
|
|
self.exists = True
|
|
self.mtime = os.stat( self.filename ).st_mtime
|
|
return True
|
|
else:
|
|
mtime = os.stat( self.filename ).st_mtime
|
|
if mtime != self.mtime:
|
|
self.mtime = mtime
|
|
return True
|
|
else:
|
|
if self.exists:
|
|
self.exists = False
|
|
return True
|
|
|
|
return False
|
|
|
|
class MonitorThread(threading.Thread):
|
|
def __init__(self, monitor):
|
|
threading.Thread.__init__ ( self )
|
|
self.monitor = monitor
|
|
|
|
def run(self):
|
|
while(1):
|
|
self.monitor.checkFiles()
|
|
time.sleep(1)
|
|
|
|
class FileMonitor(object):
|
|
def __init__( self ):
|
|
self.monitorId = 0
|
|
self.monitoredFiles = []
|
|
self.monitorThread = MonitorThread( self )
|
|
self.monitorThread.setDaemon( True )
|
|
self.monitorThread.start()
|
|
|
|
def addMonitor( self, filename, callback, args = None ):
|
|
self.monitorId += 1
|
|
self.monitoredFiles.append( _MonitoredFile( filename, callback, self.monitorId, args ) )
|
|
return self.monitorId
|
|
|
|
def removeMonitor( self, monitorId ):
|
|
for monitored in self.monitoredFiles:
|
|
if monitorId == monitored.monitorId:
|
|
self.monitoredFiles.remove( monitored )
|
|
break
|
|
|
|
def checkFiles( self ):
|
|
for monitored in self.monitoredFiles:
|
|
if monitored.hasChanged():
|
|
if monitored.args:
|
|
GLib.idle_add( monitored.callback, monitored.args )
|
|
else:
|
|
GLib.idle_add( monitored.callback )
|
|
|
|
monitor = FileMonitor()
|