mirror of
https://github.com/9001/copyparty.git
synced 2026-04-02 05:25:34 +10:00
120 lines
3.6 KiB
Python
120 lines
3.6 KiB
Python
#!/usr/bin/env python
|
|
# coding: utf-8
|
|
from __future__ import print_function, unicode_literals
|
|
|
|
import sys
|
|
import time
|
|
import threading
|
|
import multiprocessing as mp
|
|
from datetime import datetime, timedelta
|
|
import calendar
|
|
|
|
from .__init__ import PY2, WINDOWS
|
|
from .tcpsrv import TcpSrv
|
|
|
|
|
|
class SvcHub(object):
|
|
"""
|
|
Hosts all services which cannot be parallelized due to reliance on monolithic resources.
|
|
Creates a Broker which does most of the heavy stuff; hosted services can use this to perform work:
|
|
hub.broker.put(retq, action, arg1, argN).
|
|
|
|
Either BrokerThr (plain threads) or BrokerMP (multiprocessing) is used depending on configuration.
|
|
To receive any output returned by action, provide a queue-object for retq, else None.
|
|
"""
|
|
|
|
def __init__(self, args):
|
|
self.args = args
|
|
|
|
self.log_mutex = threading.Lock()
|
|
self.next_day = 0
|
|
|
|
# initiate all services to manage
|
|
self.tcpsrv = TcpSrv(self)
|
|
|
|
# decide which worker impl to use
|
|
if self.check_mp_enable():
|
|
from .broker_mp import BrokerMp as Broker
|
|
else:
|
|
self.log("root", "cannot efficiently use multiple CPU cores")
|
|
from .broker_thr import BrokerThr as Broker
|
|
|
|
self.broker = Broker(self)
|
|
|
|
def run(self):
|
|
thr = threading.Thread(target=self.tcpsrv.run)
|
|
thr.daemon = True
|
|
thr.start()
|
|
|
|
# winxp/py2.7 support: thr.join() kills signals
|
|
try:
|
|
while True:
|
|
time.sleep(9001)
|
|
except KeyboardInterrupt:
|
|
print("OPYTHAT")
|
|
self.tcpsrv.shutdown()
|
|
self.broker.shutdown()
|
|
print("nailed it")
|
|
|
|
def log(self, src, msg):
|
|
"""handles logging from all components"""
|
|
now = time.time()
|
|
if now >= self.next_day:
|
|
dt = datetime.utcfromtimestamp(now)
|
|
print("\033[36m{}\033[0m".format(dt.strftime("%Y-%m-%d")))
|
|
|
|
# unix timestamp of next 00:00:00 (leap-seconds safe)
|
|
day_now = dt.day
|
|
while dt.day == day_now:
|
|
dt += timedelta(hours=12)
|
|
|
|
dt = dt.replace(hour=0, minute=0, second=0)
|
|
self.next_day = calendar.timegm(dt.utctimetuple())
|
|
|
|
with self.log_mutex:
|
|
ts = datetime.utcfromtimestamp(now).strftime("%H:%M:%S")
|
|
print("\033[36m{} \033[33m{:21} \033[0m{}".format(ts, src, msg))
|
|
|
|
def check_mp_support(self):
|
|
vmin = sys.version_info[1]
|
|
if WINDOWS:
|
|
msg = "need python 3.3 or newer for multiprocessing;"
|
|
if PY2:
|
|
# py2 pickler doesn't support winsock
|
|
return msg
|
|
elif vmin < 3:
|
|
return msg
|
|
else:
|
|
msg = "need python 2.7 or 3.3+ for multiprocessing;"
|
|
if not PY2 and vmin < 3:
|
|
return msg
|
|
|
|
try:
|
|
x = mp.Queue(1)
|
|
x.put(["foo", "bar"])
|
|
if x.get()[0] != "foo":
|
|
raise Exception()
|
|
except:
|
|
return "multiprocessing is not supported on your platform;"
|
|
|
|
return None
|
|
|
|
def check_mp_enable(self):
|
|
if self.args.j == 0:
|
|
self.log("root", "multiprocessing disabled by argument -j 0;")
|
|
return False
|
|
|
|
try:
|
|
# support vscode debugger (bonus: same behavior as on windows)
|
|
mp.set_start_method("spawn", True)
|
|
except AttributeError:
|
|
# py2.7 probably, anyways dontcare
|
|
pass
|
|
|
|
err = self.check_mp_support()
|
|
if not err:
|
|
return True
|
|
else:
|
|
self.log("root", err)
|
|
return False
|