def GetWaitingEstimate(self, bandwidth_type, time_delta, max_allowed):
with self._lock:
if time_delta is None: # this is monthly
dt = self._GetCurrentDateTime()
(year, month) = (dt.year, dt.month)
next_month_year = year
if month == 12:
next_month_year += 1
next_month = (month % 12) + 1
next_month_dt = datetime.datetime(next_month_year, next_month,
1)
next_month_time = int(
calendar.timegm(next_month_dt.timetuple()))
return HydrusData.GetTimeDeltaUntilTime(next_month_time)
else:
# we want the highest time_delta at which usage is >= than max_allowed
# time_delta subtract that amount is the time we have to wait for usage to be less than max_allowed
# e.g. if in the past 24 hours there was a bunch of usage 16 hours ago clogging it up, we'll have to wait ~8 hours
(window,
counter) = self._GetWindowAndCounter(bandwidth_type,
time_delta)
time_delta_in_which_bandwidth_counts = time_delta + window
time_and_values = list(counter.items())
time_and_values.sort(reverse=True)
now = HydrusData.GetNow()
usage = 0
for (timestamp, value) in time_and_values:
current_search_time_delta = now - timestamp
if current_search_time_delta > time_delta_in_which_bandwidth_counts: # we are searching beyond our time delta. no need to wait
break
usage += value
if usage >= max_allowed:
return time_delta_in_which_bandwidth_counts - current_search_time_delta
return 0
def __init__(self,
parent,
share_key,
name,
text,
timeout,
hashes,
new_share=False):
Dialog.__init__(self, parent, 'configure local booru share')
self._name = QW.QLineEdit(self)
self._text = QW.QPlainTextEdit(self)
self._text.setMinimumHeight(100)
message = 'expires in'
self._timeout_number = ClientGUICommon.NoneableSpinCtrl(
self,
message,
none_phrase='no expiration',
max=1000000,
multiplier=1)
self._timeout_multiplier = ClientGUICommon.BetterChoice(self)
self._timeout_multiplier.addItem('minutes', 60)
self._timeout_multiplier.addItem('hours', 60 * 60)
self._timeout_multiplier.addItem('days', 60 * 60 * 24)
self._copy_internal_share_link = QW.QPushButton(
'copy internal share link', self)
self._copy_internal_share_link.clicked.connect(
self.EventCopyInternalShareURL)
self._copy_external_share_link = QW.QPushButton(
'copy external share link', self)
self._copy_external_share_link.clicked.connect(
self.EventCopyExternalShareURL)
self._ok = QW.QPushButton('ok', self)
self._ok.clicked.connect(self.accept)
self._ok.setObjectName('HydrusAccept')
self._cancel = QW.QPushButton('cancel', self)
self._cancel.clicked.connect(self.reject)
self._cancel.setObjectName('HydrusCancel')
#
self._share_key = share_key
self._name.setText(name)
self._text.setPlainText(text)
if timeout is None:
self._timeout_number.SetValue(None)
self._timeout_multiplier.SetValue(60)
else:
time_left = HydrusData.GetTimeDeltaUntilTime(timeout)
if time_left < 60 * 60 * 12: time_value = 60
elif time_left < 60 * 60 * 24 * 7: time_value = 60 * 60
else: time_value = 60 * 60 * 24
self._timeout_number.SetValue(time_left // time_value)
self._timeout_multiplier.SetValue(time_value)
self._hashes = hashes
self._service = HG.client_controller.services_manager.GetService(
CC.LOCAL_BOORU_SERVICE_KEY)
internal_port = self._service.GetPort()
if internal_port is None:
self._copy_internal_share_link.setEnabled(False)
self._copy_external_share_link.setEnabled(False)
#
rows = []
rows.append(('share name: ', self._name))
rows.append(('share text: ', self._text))
gridbox = ClientGUICommon.WrapInGrid(self, rows)
timeout_box = QP.HBoxLayout()
QP.AddToLayout(timeout_box, self._timeout_number,
CC.FLAGS_EXPAND_BOTH_WAYS)
QP.AddToLayout(timeout_box, self._timeout_multiplier,
CC.FLAGS_EXPAND_BOTH_WAYS)
link_box = QP.HBoxLayout()
QP.AddToLayout(link_box, self._copy_internal_share_link,
CC.FLAGS_CENTER_PERPENDICULAR)
QP.AddToLayout(link_box, self._copy_external_share_link,
CC.FLAGS_CENTER_PERPENDICULAR)
b_box = QP.HBoxLayout()
QP.AddToLayout(b_box, self._ok, CC.FLAGS_CENTER_PERPENDICULAR)
QP.AddToLayout(b_box, self._cancel, CC.FLAGS_CENTER_PERPENDICULAR)
vbox = QP.VBoxLayout()
intro = 'Sharing ' + HydrusData.ToHumanInt(len(
self._hashes)) + ' files.'
intro += os.linesep + 'Title and text are optional.'
if new_share:
intro += os.linesep + 'The link will not work until you ok this dialog.'
QP.AddToLayout(vbox, ClientGUICommon.BetterStaticText(self, intro),
CC.FLAGS_EXPAND_PERPENDICULAR)
QP.AddToLayout(vbox, gridbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR)
QP.AddToLayout(vbox, timeout_box, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR)
QP.AddToLayout(vbox, link_box, CC.FLAGS_ON_RIGHT)
QP.AddToLayout(vbox, b_box, CC.FLAGS_ON_RIGHT)
self.setLayout(vbox)
size_hint = self.sizeHint()
size_hint.setWidth(max(size_hint.width(), 350))
QP.SetInitialSize(self, size_hint)
HG.client_controller.CallAfterQtSafe(self._ok, self._ok.setFocus,
QC.Qt.OtherFocusReason)