class Client(Node):
"""Client class"""
def __init__(self, role, ip, port, uname, watch_dirs, server):
super(Client, self).__init__(role, ip, port, uname, watch_dirs)
self.server = server
self.mfiles = FilesPersistentSet(
pkl_filename='client.pkl') #set() #set of modified files
self.rfiles = set() #set of removed files
self.pulled_files = set()
self.server_available = True
def push_file(self, filename, dest_file, dest_uname, dest_ip):
"""push file 'filename' to the destination"""
# dest_file = Node.get_dest_path(filename, dest_uname)
proc = subprocess.Popen(
['scp', filename,
"%s@%s:%s" % (dest_uname, dest_ip, dest_file)])
push_status = proc.wait()
logger.debug("returned status %s", push_status)
return push_status
def pull_file(self, filename, source_uname, source_ip):
"""pull file 'filename' from the source"""
my_file = Node.get_dest_path(filename, self.username)
self.pulled_files.add(my_file)
proc = subprocess.Popen(
['scp',
"%s@%s:%s" % (source_uname, source_ip, filename), my_file])
return_status = proc.wait()
logger.debug("returned status %s", return_status)
def get_public_key(self):
"""Return public key of this client"""
pubkey = None
pubkey_dirname = os.path.join("/home", self.username, ".ssh")
logger.debug("public key directory %s", pubkey_dirname)
for tuple in os.walk(pubkey_dirname):
dirname, dirnames, filenames = tuple
break
logger.debug("public key dir files %s", filenames)
for filename in filenames:
if '.pub' in filename:
pubkey_filepath = os.path.join(dirname, filename)
logger.debug("public key file %s", pubkey_filepath)
pubkey = open(pubkey_filepath, 'r').readline()
logger.debug("public key %s", pubkey)
return pubkey
def find_modified(self):
"""Find all those files which have been modified when sync demon was not running"""
for directory in self.watch_dirs:
dirwalk = os.walk(directory)
for tuple in dirwalk:
dirname, dirnames, filenames = tuple
break
for filename in filenames:
file_path = os.path.join(dirname, filename)
logger.debug(
"checked file if modified before client was running: %s",
file_path)
mtime = os.path.getmtime(file_path)
#TODO save and restore last_synctime
if mtime > self.mfiles.get_modified_timestamp():
logger.debug("modified before client was running %s",
file_path)
self.mfiles.add(file_path, mtime)
def sync_files(self):
"""Sync all the files present in the mfiles set and push this set"""
mfiles = self.mfiles
while True:
try:
time.sleep(10)
for filedata in mfiles.list():
filename = filedata.name
logger.info("push filedata object to server %s", filedata)
server_uname, server_ip, server_port = self.server
dest_file = rpc.req_push_file(server_ip, server_port,
filedata, self.username,
self.ip, self.port)
logger.debug("destination file name %s", dest_file)
if dest_file is None:
break
push_status = self.push_file(filename, dest_file,
server_uname, server_ip)
if (push_status < 0):
break
rpc_status = rpc.ack_push_file(server_ip, server_port,
dest_file, self.username,
self.ip, self.port)
if rpc_status is None:
break
mfiles.remove(filename)
self.mfiles.update_modified_timestamp()
except KeyboardInterrupt:
break
def watch_files(self):
"""keep a watch on files present in sync directories"""
wm = WatchManager()
# watched events
mask = pyinotify.IN_DELETE | pyinotify.IN_CREATE | pyinotify.IN_MODIFY
notifier = pyinotify.Notifier(
wm, PTmp(self.mfiles, self.rfiles, self.pulled_files))
logger.debug("watched dir %s", self.watch_dirs)
for watch_dir in self.watch_dirs:
wm.add_watch(os.path.expanduser(watch_dir),
mask,
rec=False,
auto_add=True)
while True:
try:
time.sleep(5)
notifier.process_events()
if notifier.check_events():
notifier.read_events()
except KeyboardInterrupt:
notifier.stop()
break
def start_watch_thread(self):
"""Start threads to find modified files """
watch_thread = threading.Thread(target=self.watch_files)
watch_thread.start()
logger.info("Thread 'watchfiles' started ")
def mark_presence(self):
server_uname, server_ip, server_port = self.server
logger.debug("client call to mark available to the server")
rpc.mark_presence(server_ip, server_port, self.ip, self.port)
logger.debug("find modified files")
def activate(self):
""" Activate Client Node """
super(Client, self).activate()
self.start_watch_thread()
self.mark_presence()
self.find_modified()
class Client():
def __init__(self, dir, destination_ip, destination_user):
self.dir = dir
self.destination_ip = destination_ip
self.destination_user = destination_user
self.mfiles = FilesPersistentSet(pkl_filename = 'client.pkl')
self.rfiles = set()
self.pulled_files = set()
def push_pull_file(self):
push_status = sync_all(self.dir, self.destination_ip, self.destination_user)
return push_status
def find_modified(self):
dirwalk = os.walk(self.dir)
for tuple in dirwalk:
dirname, dirnames, filenames = tuple
break
for filename in filenames:
file_path = os.path.join(dirname,filename)
print "checked file if modified before client was running: %s" % file_path
mtime = os.path.getmtime(file_path)
if mtime > self.mfiles.get_modified_timestamp():
print "modified before client was running %s", file_path
self.mfiles.add(file_path, mtime)
def sync_files(self):
mfiles = self.mfiles
while True:
try:
time.sleep(10)
if len(mfiles.list())>0:
self.push_pull_file()
for filedata in mfiles.list():
filename = filedata.name
mfiles.remove(filename)
self.mfiles.update_modified_timestamp()
except KeyboardInterrupt:
break
def watch_files(self):
wm = WatchManager()
mask = pyinotify.IN_DELETE | pyinotify.IN_CREATE | pyinotify.IN_MODIFY
notifier = pyinotify.Notifier(wm, PTmp(self.mfiles, self.rfiles, self.pulled_files))
wm.add_watch(os.path.expanduser(self.dir), mask, rec=False, auto_add=True)
while True:
try:
time.sleep(5)
notifier.process_events()
if notifier.check_events():
notifier.read_events()
except KeyboardInterrupt:
notifier.stop()
break
def start_watch_thread(self):
watch_thread = threading.Thread(target=self.watch_files)
watch_thread.start()
def start_sync_thread(self):
sync_thread = threading.Thread(target=self.sync_files)
sync_thread.start()
def activate(self):
self.start_sync_thread()
self.start_watch_thread()
self.find_modified()
class Client(Node):
"""Client class"""
def __init__(self, role, ip, port, uname, watch_dirs, server):
super(Client, self).__init__(role, ip, port, uname, watch_dirs)
self.server = server
self.mfiles = FilesPersistentSet(pkl_filename = 'client.pkl') #set() #set of modified files
self.rfiles = set() #set of removed files
self.pulled_files = set()
self.server_available = True
def push_file(self, filename, dest_file, dest_uname, dest_ip):
"""push file 'filename' to the destination"""
# dest_file = Node.get_dest_path(filename, dest_uname)
proc = subprocess.Popen(['scp', filename, "%s@%s:%s" % (dest_uname, dest_ip, dest_file)])
push_status = proc.wait()
logger.debug("returned status %s", push_status)
return push_status
def pull_file(self, filename, source_uname, source_ip):
"""pull file 'filename' from the source"""
my_file = Node.get_dest_path(filename, self.username)
self.pulled_files.add(my_file)
proc = subprocess.Popen(['scp', "%s@%s:%s" % (source_uname, source_ip, filename), my_file])
return_status = proc.wait()
logger.debug("returned status %s", return_status)
def get_public_key(self):
"""Return public key of this client"""
pubkey = None
pubkey_dirname = os.path.join("/home",self.username,".ssh")
logger.debug("public key directory %s", pubkey_dirname)
for tuple in os.walk(pubkey_dirname):
dirname, dirnames, filenames = tuple
break
logger.debug("public key dir files %s", filenames)
for filename in filenames:
if '.pub' in filename:
pubkey_filepath = os.path.join(dirname, filename)
logger.debug("public key file %s", pubkey_filepath)
pubkey = open(pubkey_filepath,'r').readline()
logger.debug("public key %s", pubkey)
return pubkey
def find_modified(self):
"""Find all those files which have been modified when sync demon was not running"""
for directory in self.watch_dirs:
dirwalk = os.walk(directory)
for tuple in dirwalk:
dirname, dirnames, filenames = tuple
break
for filename in filenames:
file_path = os.path.join(dirname,filename)
logger.debug("checked file if modified before client was running: %s", file_path)
mtime = os.path.getmtime(file_path)
#TODO save and restore last_synctime
if mtime > self.mfiles.get_modified_timestamp():
logger.debug("modified before client was running %s", file_path)
self.mfiles.add(file_path, mtime)
def sync_files(self):
"""Sync all the files present in the mfiles set and push this set"""
mfiles = self.mfiles
while True:
try:
time.sleep(10)
for filedata in mfiles.list():
filename = filedata.name
logger.info("push filedata object to server %s" , filedata)
server_uname, server_ip, server_port = self.server
dest_file = rpc.req_push_file(server_ip, server_port, filedata, self.username, self.ip, self.port)
logger.debug("destination file name %s", dest_file)
if dest_file is None:
break
push_status = self.push_file(filename, dest_file, server_uname, server_ip)
if (push_status < 0):
break
rpc_status = rpc.ack_push_file(server_ip, server_port, dest_file, self.username, self.ip, self.port)
if rpc_status is None:
break
mfiles.remove(filename)
self.mfiles.update_modified_timestamp()
except KeyboardInterrupt:
break
def watch_files(self):
"""keep a watch on files present in sync directories"""
wm = WatchManager()
# watched events
mask = pyinotify.IN_DELETE | pyinotify.IN_CREATE | pyinotify.IN_MODIFY
notifier = pyinotify.Notifier(wm, PTmp(self.mfiles, self.rfiles, self.pulled_files))
logger.debug("watched dir %s",self.watch_dirs)
for watch_dir in self.watch_dirs:
wm.add_watch(os.path.expanduser(watch_dir), mask, rec=False, auto_add=True)
while True:
try:
time.sleep(5)
notifier.process_events()
if notifier.check_events():
notifier.read_events()
except KeyboardInterrupt:
notifier.stop()
break
def start_watch_thread(self):
"""Start threads to find modified files """
watch_thread = threading.Thread(target=self.watch_files)
watch_thread.start()
logger.info("Thread 'watchfiles' started ")
def mark_presence(self):
server_uname, server_ip, server_port = self.server
logger.debug("client call to mark available to the server")
rpc.mark_presence(server_ip, server_port, self.ip, self.port)
logger.debug("find modified files")
def activate(self):
""" Activate Client Node """
super(Client, self).activate()
self.start_watch_thread()
self.mark_presence()
self.find_modified()