Spaces:
Build error
Build error
File size: 2,050 Bytes
98bb602 |
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 |
import time
import threading
import subprocess
from typing import List, Union
class Channel:
def __init__(self, source, destination, sync_deletions=False, every=60, exclude: Union[str, List, None] = None):
self.source = source
self.destination = destination
self.event = threading.Event()
self.syncing_thread = threading.Thread(target=self._sync, args=())
self.sync_deletions = sync_deletions
self.every = every
if not exclude: exclude = []
if isinstance(exclude,str): exclude = [exclude]
self.exclude = exclude
self.command = ['rsync', '-aP']
def alive(self):
if self.syncing_thread.is_alive(): return True
else: return False
def _sync(self):
command = self.command
for exclusion in self.exclude:
command.append(f'--exclude={exclusion}')
command.extend([f'{self.source}/', f'{self.destination}/'])
if self.sync_deletions: command.append('--delete')
while not self.event.is_set():
subprocess.run(command)
time.sleep(self.every)
def copy(self):
command = self.command
for exclusion in self.exclude:
command.append(f'--exclude={exclusion}')
command.extend([f'{self.source}/', f'{self.destination}/'])
if self.sync_deletions: command.append('--delete')
subprocess.run(command)
return True
def start(self):
if self.syncing_thread.is_alive():
self.event.set()
self.syncing_thread.join()
if self.event.is_set(): self.event.clear()
if self.syncing_thread._started.is_set(): self.syncing_thread = threading.Thread(target=self._sync, args=())
self.syncing_thread.start()
return self.alive()
def stop(self):
if self.alive():
self.event.set()
self.syncing_thread.join()
while self.alive():
if not self.alive(): break
return not self.alive() |