You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
calculate-utils-3-lib/pym/calculate/lib/utils/binhosts.py

132 lines
5.2 KiB

# -*- coding: utf-8 -*-
# Copyright 2015-2016 Mir Calculate. http://www.calculate-linux.org
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import urllib2
import re
import time
from os import path
from calculate.lib.utils.tools import SingletonParam
class BinhostsBase(object):
def __init__(self, timeout, revision_path, ts_path, last_ts, binhost_list):
self.timeout = int(timeout)
self.revision_path = revision_path
self.ts_path = ts_path
self.last_ts = last_ts
self.binhost_list = binhost_list
self.data = None
self.binhosts_data = {}
def check_binhost(self, binhost):
if binhost not in self.binhosts_data:
revision_files = [path.join(binhost, x)
for x in self.revision_path]
try:
data = None
for fn in revision_files:
if data is None:
data = urllib2.urlopen(fn, timeout=self.timeout).read()
elif data != urllib2.urlopen(fn,
timeout=self.timeout).read():
res = ""
break
else:
res = data
except urllib2.URLError:
res = ""
except BaseException as e:
if isinstance(e, KeyboardInterrupt):
raise
res = ""
self.binhosts_data[binhost] = res
return self.binhosts_data[binhost]
re_revison = re.compile("\w+=(\w+)")
def get_timestamp(self, binhost):
"""
Возвращает таймстам полученный от сервера
"""
DAY = 60 * 60 * 24
timestamp_file = path.join(binhost, self.ts_path)
try:
t = time.time()
data = urllib2.urlopen(timestamp_file,
timeout=self.timeout).read().strip()
if data.isdigit() and self.last_ts.isdigit():
return (data, int((time.time() - t) * 1000),
t - int(data) < 5 * DAY,
int(data) < int(self.last_ts))
except urllib2.URLError as e:
pass
except BaseException as e:
if isinstance(e, KeyboardInterrupt):
raise
pass
return "", -1, False, True
def generate_by_timestamp(self):
for host in self.binhost_list:
if host:
ts_content, duration, good, downgrade = \
self.get_timestamp(host)
if ts_content:
yield host, ts_content, str(duration), good, downgrade
def get_sorted(self):
if self.data is None:
self.data = []
for host, ts, t, good, downgrade in sorted(
self.generate_by_timestamp(),
# критерий сортировки между серверами
# актуальные сортируются по скорости
# неактульные по timestamp
# актуальность - 5 дней
reverse=True,
key=lambda x: ( # приоритетны актуальные (не более 5 дней)
x[3],
# приоритет для неактуальных
# самое свежее
0 if x[3] else int(x[1]),
# приоритет для неактуальных
# самое быстрое
0 if x[3] else -int(x[2]),
# приоритет для актуальных
# самое быстрое
-int(x[2]) if x[3] else 0,
# самое свежее
int(x[1]) if x[3] else 0)):
# rev_data = self.check_binhost(host)
self.data.append([host, ts, t, good, downgrade])
return self.data
def is_cache(self):
return self.data is not None
@classmethod
def param_id(cls, *args, **kw):
if not kw:
return ",".join(str(x) for x in args)
else:
return "%s|%s" % (",".join(str(x) for x in args),
",".join("%s:%s" % (str(k), str(v)) for x in
kw.items()))
class Binhosts(BinhostsBase):
__metaclass__ = SingletonParam