本文主要是介绍腾讯云被当做肉鸡了,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
腾讯云主机被黑之后装了hadoop,估计是被肉鸡,做了计算了,发现cpu基本都在99%。如下是py脚本,不知道是干啥的…
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright 2012-2016 Matt Martz
# Reedited By CoDeX.
# All Rights Reserved.
#
# 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 os
import re
import csv
import sys
import math
import errno
import signal
import socket
import timeit
import datetime
import platform
import threading
import xml.parsers.expattry:import gzipGZIP_BASE = gzip.GzipFile
except ImportError:gzip = NoneGZIP_BASE = object__version__ = '1.0.6'class FakeShutdownEvent(object):"""Class to fake a threading.Event.isSet so that users of this moduleare not required to register their own threading.Event()"""@staticmethoddef isSet():"Dummy method to always return false"""return False# Some global variables we use
USER_AGENT = None
SOURCE = None
SHUTDOWN_EVENT = FakeShutdownEvent()
SCHEME = 'http'
DEBUG = False# Used for bound_interface
SOCKET_SOCKET = socket.socket# Begin import game to handle Python 2 and Python 3
try:import json
except ImportError:try:import simplejson as jsonexcept ImportError:json = Nonetry:import xml.etree.cElementTree as ET
except ImportError:try:import xml.etree.ElementTree as ETexcept ImportError:from xml.dom import minidom as DOMET = Nonetry:from urllib2 import urlopen, Request, HTTPError, URLError
except ImportError:from urllib.request import urlopen, Request, HTTPError, URLErrortry:from httplib import HTTPConnection
except ImportError:from http.client import HTTPConnectiontry:from httplib import HTTPSConnection
except ImportError:try:from http.client import HTTPSConnectionexcept ImportError:HTTPSConnection = Nonetry:from Queue import Queue
except ImportError:from queue import Queuetry:from urlparse import urlparse
except ImportError:from urllib.parse import urlparsetry:from urlparse import parse_qs
except ImportError:try:from urllib.parse import parse_qsexcept ImportError:from cgi import parse_qstry:from hashlib import md5
except ImportError:from md5 import md5try:from argparse import ArgumentParser as ArgParserfrom argparse import SUPPRESS as ARG_SUPPRESSPARSER_TYPE_INT = intPARSER_TYPE_STR = str
except ImportError:from optparse import OptionParser as ArgParserfrom optparse import SUPPRESS_HELP as ARG_SUPPRESSPARSER_TYPE_INT = 'int'PARSER_TYPE_STR = 'string'try:from cStringIO import StringIOBytesIO = None
except ImportError:try:from StringIO import StringIOBytesIO = Noneexcept ImportError:from io import StringIO, BytesIOtry:import __builtin__
except ImportError:import builtinsfrom io import TextIOWrapper, FileIOclass _Py3Utf8Stdout(TextIOWrapper):"""UTF-8 encoded wrapper around stdout for py3, to overrideASCII stdout"""def __init__(self, **kwargs):buf = FileIO(sys.stdout.fileno(), 'w')super(_Py3Utf8Stdout, self).__init__(buf,encoding='utf8',errors='strict')def write(self, s):super(_Py3Utf8Stdout, self).write(s)self.flush()_py3_print = getattr(builtins, 'print')_py3_utf8_stdout = _Py3Utf8Stdout()def to_utf8(v):"""No-op encode to utf-8 for py3"""return vdef print_(*args, **kwargs):"""Wrapper function for py3 to print, with a utf-8 encoded stdout"""kwargs['file'] = _py3_utf8_stdout_py3_print(*args, **kwargs)
else:del __builtin__def to_utf8(v):"""Encode value to utf-8 if possible for py2"""try:return v.encode('utf8', 'strict')except AttributeError:return vdef print_(*args, **kwargs):"""The new-style print function for Python 2.4 and 2.5.Taken from https://pypi.python.org/pypi/six/Modified to set encoding to UTF-8 always"""fp = kwargs.pop("file", sys.stdout)if fp is None:returndef write(data):if not isinstance(data, basestring):data = str(data)# If the file has an encoding, encode unicode with it.encoding = 'utf8' # Always trust UTF-8 for outputif (isinstance(fp, file) andisinstance(data, unicode) andencoding is not None):errors = getattr(fp, "errors", None)if errors is None:errors = "strict"data = data.encode(encoding, errors)fp.write(data)want_unicode = Falsesep = kwargs.pop("sep", None)if sep is not None:if isinstance(sep, unicode):want_unicode = Trueelif not isinstance(sep, str):raise TypeError("sep must be None or a string")end = kwargs.pop("end", None)if end is not None:if isinstance(end, unicode):want_unicode = Trueelif not isinstance(end, str):raise TypeError("end must be None or a string")if kwargs:raise TypeError("invalid keyword arguments to print()")if not want_unicode:for arg in args:if isinstance(arg, unicode):want_unicode = Truebreakif want_unicode:newline = unicode("\n")space = unicode(" ")else:newline = "\n"space = " "if sep is None:sep = spaceif end is None:end = newlinefor i, arg in enumerate(args):if i:write(sep)write(arg)write(end)# Exception "constants" to support Python 2 through Python 3
try:import ssltry:CERT_ERROR = (ssl.CertificateError,)except AttributeError:CERT_ERROR = tuple()HTTP_ERRORS = ((HTTPError, URLError, socket.error, ssl.SSLError) +CERT_ERROR)
except ImportError:HTTP_ERRORS = (HTTPError, URLError, socket.error)class SpeedtestException(Exception):"""Base exception for this module"""class SpeedtestCLIError(SpeedtestException):"""Generic exception for raising errors during CLI operation"""class SpeedtestHTTPError(SpeedtestException):"""Base HTTP exception for this module"""class SpeedtestConfigError(SpeedtestException):"""Configuration provided is invalid"""class ConfigRetrievalError(SpeedtestHTTPError):"""Could not retrieve config.php"""class ServersRetrievalError(SpeedtestHTTPError):"""Could not retrieve speedtest-servers.php"""class InvalidServerIDType(SpeedtestException):"""Server ID used for filtering was not an integer"""class NoMatchedServers(SpeedtestException):"""No servers matched when filtering"""class SpeedtestMiniConnectFailure(SpeedtestException):"""Could not connect to the provided speedtest mini server"""class InvalidSpeedtestMiniServer(SpeedtestException):"""Server provided as a speedtest mini server does not actually appearto be a speedtest mini server"""class ShareResultsConnectFailure(SpeedtestException):"""Could not connect to speedtest.net API to POST results"""class ShareResultsSubmitFailure(SpeedtestException):"""Unable to successfully POST results to speedtest.net API afterconnection"""class SpeedtestUploadTimeout(SpeedtestException):"""testlength configuration reached during uploadUsed to ensure the upload halts when no additional data should be sent"""class SpeedtestBestServerFailure(SpeedtestException):"""Unable to determine best server"""class GzipDecodedResponse(GZIP_BASE):"""A file-like object to decode a response encoded with the gzipmethod, as described in RFC 1952.Largely copied from ``xmlrpclib``/``xmlrpc.client`` and modifiedto work for py2.4-py3"""def __init__(self, response):# response doesn't support tell() and read(), required by# GzipFileif not gzip:raise SpeedtestHTTPError('HTTP response body is gzip encoded, ''but gzip support is not available')IO = BytesIO or StringIOself.io = IO()while 1:chunk = response.read(1024)if len(chunk) == 0:breakself.io.write(chunk)self.io.seek(0)gzip.GzipFile.__init__(self, mode='rb', fileobj=self.io)def close(self):try:gzip.GzipFile.close(self)finally:self.io.close()def get_exception():"""Helper function to work with py2.4-py3 for getting the currentexception in a try/except block"""return sys.exc_info()[1]def bound_socket(*args, **kwargs):"""Bind socket to a specified source IP address"""sock = SOCKET_SOCKET(*args, **kwargs)sock.bind((SOURCE, 0))return sockdef distance(origin, destination):"""Determine distance between 2 sets of [lat,lon] in km"""lat1, lon1 = originlat2, lon2 = destinationradius = 6371 # kmdlat = math.radians(lat2 - lat1)dlon = math.radians(lon2 - lon1)a = (math.sin(dlat / 2) * math.sin(dlat / 2) +math.cos(math.radians(lat1)) *math.cos(math.radians(lat2)) * math.sin(dlon / 2) *math.sin(dlon / 2))c = 2 * math.atan2(math.sqrt(a), math.sqrt(1 - a))d = radius * creturn ddef build_user_agent():"""Build a Mozilla/5.0 compatible User-Agent string"""global USER_AGENTif USER_AGENT:return USER_AGENTua_tuple = ('Mozilla/5.0','(%s; U; %s; en-us)' % (platform.system(), platform.architecture()[0]),'Python/%s' % platform.python_version(),'(KHTML, like Gecko)','speedtest-cli/%s' % __version__)USER_AGENT = ' '.join(ua_tuple)printer(USER_AGENT, debug=True)return USER_AGENTdef build_request(url, data=None, headers=None, bump=''):"""Build a urllib2 request objectThis function automatically adds a User-Agent header to all requests"""if not USER_AGENT:build_user_agent()if not headers:headers = {}if url[0] == ':':schemed_url = '%s%s' % (SCHEME, url)else:schemed_url = urlif '?' in url:delim = '&'else:delim = '?'# WHO YOU GONNA CALL? CACHE BUSTERS!final_url = '%s%sx=%s.%s' % (schemed_url, delim,int(timeit.time.time() * 1000),bump)headers.update({'User-Agent': USER_AGENT,'Cache-Control': 'no-cache',})printer('%s %s' % (('GET', 'POST')[bool(data)], final_url),debug=True)return Request(final_url, data=data, headers=headers)def catch_request(request):"""Helper function to catch common exceptions encountered whenestablishing a connection with a HTTP/HTTPS request"""try:uh = urlopen(request)return uh, Falseexcept HTTP_ERRORS:e = get_exception()return None, edef get_response_stream(response):"""Helper function to return either a Gzip reader if``Content-Encoding`` is ``gzip`` otherwise the response itself"""try:getheader = response.headers.getheaderexcept AttributeError:getheader = response.getheaderif getheader('content-encoding') == 'gzip':return GzipDecodedResponse(response)return responsedef get_attributes_by_tag_name(dom, tag_name):"""Retrieve an attribute from an XML document and return it in aconsistent formatOnly used with xml.dom.minidom, which is likely only to be usedwith python versions older than 2.5"""elem = dom.getElementsByTagName(tag_name)[0]return dict(list(elem.attributes.items()))def print_dots(current, total, start=False, end=False):"""Built in callback function used by Thread classes for printingstatus"""if SHUTDOWN_EVENT.isSet():returnsys.stdout.write('')if current + 1 == total and end is True:sys.stdout.write('\n')sys.stdout.flush()def do_nothing(*args, **kwargs):passclass HTTPDownloader(threading.Thread):"""Thread class for retrieving a URL"""def __init__(self, i, request, start, timeout):threading.Thread.__init__(self)self.request = requestself.result = [0]self.starttime = startself.timeout = timeoutself.i = idef run(self):try:if (timeit.default_timer() - self.starttime) <= self.timeout:f = urlopen(self.request)while (not SHUTDOWN_EVENT.isSet() and(timeit.default_timer() - self.starttime) <=self.timeout):self.result.append(len(f.read(10240)))if self.result[-1] == 0:breakf.close()except IOError:passclass HTTPUploaderData(object):"""File like object to improve cutting off the upload once the timeouthas been reached"""def __init__(self, length, start, timeout):self.length = lengthself.start = startself.timeout = timeoutself._data = Noneself.total = [0]def pre_allocate(self):chars = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ'multiplier = int(round(int(self.length) / 36.0))IO = BytesIO or StringIOself._data = IO(('content1=%s' %(chars * multiplier)[0:int(self.length) - 9]).encode())@propertydef data(self):if not self._data:self.pre_allocate()return self._datadef read(self, n=10240):if ((timeit.default_timer() - self.start) <= self.timeout andnot SHUTDOWN_EVENT.isSet()):chunk = self.data.read(n)self.total.append(len(chunk))return chunkelse:raise SpeedtestUploadTimeout()def __len__(self):return self.lengthclass HTTPUploader(threading.Thread):"""Thread class for putting a URL"""def __init__(self, i, request, start, size, timeout):threading.Thread.__init__(self)self.request = requestself.request.data.start = self.starttime = startself.size = sizeself.result = Noneself.timeout = timeoutself.i = idef run(self):request = self.requesttry:if ((timeit.default_timer() - self.starttime) <= self.timeout andnot SHUTDOWN_EVENT.isSet()):try:f = urlopen(request)except TypeError:# PY24 expects a string or buffer# This also causes issues with Ctrl-C, but we will concede# for the moment that Ctrl-C on PY24 isn't immediaterequest = build_request(self.request.get_full_url(),data=request.data.read(self.size))f = urlopen(request)f.read(11)f.close()self.result = sum(self.request.data.total)else:self.result = 0except (IOError, SpeedtestUploadTimeout):self.result = sum(self.request.data.total)class SpeedtestResults(object):"""Class for holding the results of a speedtest, including:Download speedUpload speedPing/Latency to test serverData about server that the test was run againstAdditionally this class can return a result data as a dictionary or CSV,as well as submit a POST of the result data to the speedtest.net APIto get a share results image link."""def __init__(self, download=0, upload=0, ping=0, server=None):self.download = downloadself.upload = uploadself.ping = pingif server is None:self.server = {}else:self.server = serverself._share = Noneself.timestamp = '%sZ' % datetime.datetime.utcnow().isoformat()self.bytes_received = 0self.bytes_sent = 0def __repr__(self):return repr(self.dict())def share(self):"""POST data to the speedtest.net API to obtain a share resultslink"""if self._share:return self._sharedownload = int(round(self.download / 1000.0, 0))ping = int(round(self.ping, 0))upload = int(round(self.upload / 1000.0, 0))# Build the request to send results back to speedtest.net# We use a list instead of a dict because the API expects parameters# in a certain orderapi_data = ['recommendedserverid=%s' % self.server['id'],'ping=%s' % ping,'screenresolution=','promo=','download=%s' % download,'screendpi=','upload=%s' % upload,'testmethod=http','hash=%s' % md5(('%s-%s-%s-%s' %(ping, upload, download, '297aae72')).encode()).hexdigest(),'touchscreen=none','startmode=pingselect','accuracy=1','bytesreceived=%s' % self.bytes_received,'bytessent=%s' % self.bytes_sent,'serverid=%s' % self.server['id'],]headers = {'Referer': 'http://c.speedtest.net/flash/speedtest.swf'}request = build_request('://www.speedtest.net/api/api.php',data='&'.join(api_data).encode(),headers=headers)f, e = catch_request(request)if e:raise ShareResultsConnectFailure(e)response = f.read()code = f.codef.close()if int(code) != 200:raise ShareResultsSubmitFailure('Could not submit results to ''speedtest.net')qsargs = parse_qs(response.decode())resultid = qsargs.get('resultid')if not resultid or len(resultid) != 1:raise ShareResultsSubmitFailure('Could not submit results to ''speedtest.net')self._share = 'http://www.speedtest.net/result/%s.png' % resultid[0]return self._sharedef dict(self):"""Return dictionary of result data"""return {'download': self.download,'upload': self.upload,'ping': self.ping,'server': self.server,'timestamp': self.timestamp,'bytes_sent': self.bytes_sent,'bytes_received': self.bytes_received,'share': self._share,}def csv(self, delimiter=','):"""Return data in CSV format"""data = self.dict()out = StringIO()writer = csv.writer(out, delimiter=delimiter, lineterminator='')row = [data['server']['id'], data['server']['sponsor'],data['server']['name'], data['timestamp'],data['server']['d'], data['ping'], data['download'],data['upload']]writer.writerow([to_utf8(v) for v in row])return out.getvalue()def json(self, pretty=False):"""Return data in JSON format"""kwargs = {}if pretty:kwargs.update({'indent': 4,'sort_keys': True})return json.dumps(self.dict(), **kwargs)class Speedtest(object):"""Class for performing standard speedtest.net testing operations"""def __init__(self, config=None):self.config = {}self.get_config()if config is not None:self.config.update(config)self.servers = {}self.closest = []self.best = {}self.results = SpeedtestResults()def get_config(self):"""Download the speedtest.net configuration and return only the datawe are interested in"""headers = {}if gzip:headers['Accept-Encoding'] = 'gzip'request = build_request('://www.speedtest.net/speedtest-config.php',headers=headers)uh, e = catch_request(request)if e:raise ConfigRetrievalError(e)configxml = []stream = get_response_stream(uh)while 1:configxml.append(stream.read(1024))if len(configxml[-1]) == 0:breakstream.close()uh.close()if int(uh.code) != 200:return Noneprinter(''.encode().join(configxml), debug=True)try:root = ET.fromstring(''.encode().join(configxml))server_config = root.find('server-config').attribdownload = root.find('download').attribupload = root.find('upload').attrib# times = root.find('times').attribclient = root.find('client').attribexcept AttributeError:root = DOM.parseString(''.join(configxml))server_config = get_attributes_by_tag_name(root, 'server-config')download = get_attributes_by_tag_name(root, 'download')upload = get_attributes_by_tag_name(root, 'upload')# times = get_attributes_by_tag_name(root, 'times')client = get_attributes_by_tag_name(root, 'client')ignore_servers = list(map(int, server_config['ignoreids'].split(',')))ratio = int(upload['ratio'])upload_max = int(upload['maxchunkcount'])up_sizes = [32768, 65536, 131072, 262144, 524288, 1048576, 7340032]sizes = {'upload': up_sizes[ratio - 1:],'download': [350, 500, 750, 1000, 1500, 2000, 2500,3000, 3500, 4000]}size_count = len(sizes['upload'])upload_count = int(math.ceil(upload_max / size_count))counts = {'upload': upload_count,'download': int(download['threadsperurl'])}threads = {'upload': int(upload['threads']),'download': int(server_config['threadcount']) * 2}length = {'upload': int(upload['testlength']),'download': int(download['testlength'])}self.config.update({'client': client,'ignore_servers': ignore_servers,'sizes': sizes,'counts': counts,'threads': threads,'length': length,'upload_max': upload_count * size_count})self.lat_lon = (float(client['lat']), float(client['lon']))printer(self.config, debug=True)return self.configdef get_servers(self, servers=None):"""Retrieve a the list of speedtest.net servers, optionally filteredto servers matching those specified in the ``servers`` argument"""if servers is None:servers = []self.servers.clear()for i, s in enumerate(servers):try:servers[i] = int(s)except ValueError:raise InvalidServerIDType('%s is an invalid server type, must ''be int' % s)urls = ['://www.speedtest.net/speedtest-servers-static.php','http://c.speedtest.net/speedtest-servers-static.php','://www.speedtest.net/speedtest-servers.php','http://c.speedtest.net/speedtest-servers.php',]headers = {}if gzip:headers['Accept-Encoding'] = 'gzip'errors = []for url in urls:try:request = build_request('%s?threads=%s' %(url,self.config['threads']['download']),headers=headers)uh, e = catch_request(request)if e:errors.append('%s' % e)raise ServersRetrievalError()stream = get_response_stream(uh)serversxml = []while 1:serversxml.append(stream.read(1024))if len(serversxml[-1]) == 0:breakstream.close()uh.close()if int(uh.code) != 200:raise ServersRetrievalError()printer(''.encode().join(serversxml), debug=True)try:try:root = ET.fromstring(''.encode().join(serversxml))elements = root.getiterator('server')except AttributeError:root = DOM.parseString(''.join(serversxml))elements = root.getElementsByTagName('server')except (SyntaxError, xml.parsers.expat.ExpatError):raise ServersRetrievalError()for server in elements:try:attrib = server.attribexcept AttributeError:attrib = dict(list(server.attributes.items()))if servers and int(attrib.get('id')) not in servers:continueif int(attrib.get('id')) in self.config['ignore_servers']:continuetry:d = distance(self.lat_lon,(float(attrib.get('lat')),float(attrib.get('lon'))))except:continueattrib['d'] = dtry:self.servers[d].append(attrib)except KeyError:self.servers[d] = [attrib]printer(''.encode().join(serversxml), debug=True)breakexcept ServersRetrievalError:continueif servers and not self.servers:raise NoMatchedServers()return self.serversdef set_mini_server(self, server):"""Instead of querying for a list of servers, set a link to aspeedtest mini server"""urlparts = urlparse(server)name, ext = os.path.splitext(urlparts[2])if ext:url = os.path.dirname(server)else:url = serverrequest = build_request(url)uh, e = catch_request(request)if e:raise SpeedtestMiniConnectFailure('Failed to connect to %s' %server)else:text = uh.read()uh.close()extension = re.findall('upload_?[Ee]xtension: "([^"]+)"',text.decode())if not extension:for ext in ['php', 'asp', 'aspx', 'jsp']:try:f = urlopen('%s/speedtest/upload.%s' % (url, ext))except:passelse:data = f.read().strip().decode()if (f.code == 200 andlen(data.splitlines()) == 1 andre.match('size=[0-9]', data)):extension = [ext]breakif not urlparts or not extension:raise InvalidSpeedtestMiniServer('Invalid Speedtest Mini Server: ''%s' % server)self.servers = [{'sponsor': 'Speedtest Mini','name': urlparts[1],'d': 0,'url': '%s/speedtest/upload.%s' % (url.rstrip('/'), extension[0]),'latency': 0,'id': 0}]return self.serversdef get_closest_servers(self, limit=5):"""Limit servers to the closest speedtest.net servers based ongeographic distance"""if not self.servers:self.get_servers()for d in sorted(self.servers.keys()):for s in self.servers[d]:self.closest.append(s)if len(self.closest) == limit:breakelse:continuebreakprinter(self.closest, debug=True)return self.closestdef get_best_server(self, servers=None):"""Perform a speedtest.net "ping" to determine which speedtest.netserver has the lowest latency"""if not servers:if not self.closest:servers = self.get_closest_servers()servers = self.closestresults = {}for server in servers:cum = []url = os.path.dirname(server['url'])urlparts = urlparse('%s/latency.txt' % url)printer('%s %s/latency.txt' % ('GET', url), debug=True)for _ in range(0, 3):try:if urlparts[0] == 'https':h = HTTPSConnection(urlparts[1])else:h = HTTPConnection(urlparts[1])headers = {'User-Agent': USER_AGENT}start = timeit.default_timer()h.request("GET", urlparts[2], headers=headers)r = h.getresponse()total = (timeit.default_timer() - start)except HTTP_ERRORS:e = get_exception()printer('%r' % e, debug=True)cum.append(3600)continuetext = r.read(9)if int(r.status) == 200 and text == 'test=test'.encode():cum.append(total)else:cum.append(3600)h.close()avg = round((sum(cum) / 6) * 1000.0, 3)results[avg] = servertry:fastest = sorted(results.keys())[0]except IndexError:raise SpeedtestBestServerFailure('Unable to connect to servers to ''test latency.')best = results[fastest]best['latency'] = fastestself.results.ping = fastestself.results.server = bestself.best.update(best)printer(best, debug=True)return bestdef download(self, callback=do_nothing):"""Test download speed against speedtest.net"""urls = []for size in self.config['sizes']['download']:for _ in range(0, self.config['counts']['download']):urls.append('%s/random%sx%s.jpg' %(os.path.dirname(self.best['url']), size, size))request_count = len(urls)requests = []for i, url in enumerate(urls):requests.append(build_request(url, bump=i))def producer(q, requests, request_count):for i, request in enumerate(requests):thread = HTTPDownloader(i, request, start,self.config['length']['download'])thread.start()q.put(thread, True)callback(i, request_count, start=True)finished = []def consumer(q, request_count):while len(finished) < request_count:thread = q.get(True)while thread.isAlive():thread.join(timeout=0.1)finished.append(sum(thread.result))callback(thread.i, request_count, end=True)q = Queue(self.config['threads']['download'])prod_thread = threading.Thread(target=producer,args=(q, requests, request_count))cons_thread = threading.Thread(target=consumer,args=(q, request_count))start = timeit.default_timer()prod_thread.start()cons_thread.start()while prod_thread.isAlive():prod_thread.join(timeout=0.1)while cons_thread.isAlive():cons_thread.join(timeout=0.1)stop = timeit.default_timer()self.results.bytes_received = sum(finished)self.results.download = ((self.results.bytes_received / (stop - start)) * 8.0)if self.results.download > 100000:self.config['threads']['upload'] = 8return self.results.downloaddef upload(self, callback=do_nothing, pre_allocate=True):"""Test upload speed against speedtest.net"""sizes = []for size in self.config['sizes']['upload']:for _ in range(0, self.config['counts']['upload']):sizes.append(size)# request_count = len(sizes)request_count = self.config['upload_max']requests = []for i, size in enumerate(sizes):# We set ``0`` for ``start`` and handle setting the actual# ``start`` in ``HTTPUploader`` to get better measurementsdata = HTTPUploaderData(size, 0, self.config['length']['upload'])if pre_allocate:data.pre_allocate()requests.append((build_request(self.best['url'], data),size))def producer(q, requests, request_count):for i, request in enumerate(requests[:request_count]):thread = HTTPUploader(i, request[0], start, request[1],self.config['length']['upload'])thread.start()q.put(thread, True)callback(i, request_count, start=True)finished = []def consumer(q, request_count):while len(finished) < request_count:thread = q.get(True)while thread.isAlive():thread.join(timeout=0.1)finished.append(thread.result)callback(thread.i, request_count, end=True)q = Queue(self.config['threads']['upload'])prod_thread = threading.Thread(target=producer,args=(q, requests, request_count))cons_thread = threading.Thread(target=consumer,args=(q, request_count))start = timeit.default_timer()prod_thread.start()cons_thread.start()while prod_thread.isAlive():prod_thread.join(timeout=0.1)while cons_thread.isAlive():cons_thread.join(timeout=0.1)stop = timeit.default_timer()self.results.bytes_sent = sum(finished)self.results.upload = ((self.results.bytes_sent / (stop - start)) * 8.0)return self.results.uploaddef ctrl_c(signum, frame):"""Catch Ctrl-C key sequence and set a SHUTDOWN_EVENT for our threadedoperations"""SHUTDOWN_EVENT.set()print_('\nCancelling...')sys.exit(0)def version():"""Print the version"""print_(__version__)sys.exit(0)def csv_header():"""Print the CSV Headers"""print_('Server ID,Sponsor,Server Name,Timestamp,Distance,Ping,Download,''Upload')sys.exit(0)def parse_args():"""Function to handle building and parsing of command line arguments"""description = ('Command line interface for testing internet bandwidth using ''speedtest.net.\n''------------------------------------------------------------''--------------\n''https://github.com/sivel/speedtest-cli')parser = ArgParser(description=description)# Give optparse.OptionParser an `add_argument` method for# compatibility with argparse.ArgumentParsertry:parser.add_argument = parser.add_optionexcept AttributeError:passparser.add_argument('--no-download', dest='download', default=True,action='store_const', const=False,help='Do not perform download test')parser.add_argument('--no-upload', dest='upload', default=True,action='store_const', const=False,help='Do not perform upload test')parser.add_argument('--bytes', dest='units', action='store_const',const=('byte', 8), default=('bit', 1),help='Display values in bytes instead of bits. Does ''not affect the image generated by --share, nor ''output from --json or --csv')parser.add_argument('--share', action='store_true',help='Generate and provide a URL to the speedtest.net ''share results image, not displayed with --csv')parser.add_argument('--simple', action='store_true', default=False,help='Suppress verbose output, only show basic ''information')parser.add_argument('--csv', action='store_true', default=False,help='Suppress verbose output, only show basic ''information in CSV format. Speeds listed in ''bit/s and not affected by --bytes')parser.add_argument('--csv-delimiter', default=',', type=PARSER_TYPE_STR,help='Single character delimiter to use in CSV ''output. Default ","')parser.add_argument('--csv-header', action='store_true', default=False,help='Print CSV headers')parser.add_argument('--json', action='store_true', default=False,help='Suppress verbose output, only show basic ''information in JSON format. Speeds listed in ''bit/s and not affected by --bytes')parser.add_argument('--list', action='store_true',help='Display a list of speedtest.net servers ''sorted by distance')parser.add_argument('--server', help='Specify a server ID to test against',type=PARSER_TYPE_INT)parser.add_argument('--mini', help='URL of the Speedtest Mini server')parser.add_argument('--source', help='Source IP address to bind to')parser.add_argument('--timeout', default=10, type=PARSER_TYPE_INT,help='HTTP timeout in seconds. Default 10')parser.add_argument('--secure', action='store_true',help='Use HTTPS instead of HTTP when communicating ''with speedtest.net operated servers')parser.add_argument('--no-pre-allocate', dest='pre_allocate',action='store_const', default=True, const=False,help='Do not pre allocate upload data. Pre allocation ''is enabled by default to improve upload ''performance. To support systems with ''insufficient memory, use this option to avoid a ''MemoryError')parser.add_argument('--version', action='store_true',help='Show the version number and exit')parser.add_argument('--debug', action='store_true',help=ARG_SUPPRESS, default=ARG_SUPPRESS)options = parser.parse_args()if isinstance(options, tuple):args = options[0]else:args = optionsreturn argsdef validate_optional_args(args):"""Check if an argument was provided that depends on a module that maynot be part of the Python standard library.If such an argument is supplied, and the module does not exist, exitwith an error stating which module is missing."""optional_args = {'json': ('json/simplejson python module', json),'secure': ('SSL support', HTTPSConnection),}for arg, info in optional_args.items():if getattr(args, arg, False) and info[1] is None:raise SystemExit('%s is not installed. --%s is ''unavailable' % (info[0], arg))def printer(string, quiet=False, debug=False, **kwargs):"""Helper function to print a string only when not quiet"""if debug and not DEBUG:returnif debug:out = '\033[1;30mDEBUG: %s\033[0m' % stringelse:out = stringif not quiet:print_(out, **kwargs)def shell():"""Run the full speedtest.net test"""global SHUTDOWN_EVENT, SOURCE, SCHEME, DEBUGSHUTDOWN_EVENT = threading.Event()signal.signal(signal.SIGINT, ctrl_c)args = parse_args()# Print the version and exitif args.version:version()if not args.download and not args.upload:raise SpeedtestCLIError('Cannot supply both --no-download and ''--no-upload')if args.csv_header:csv_header()if len(args.csv_delimiter) != 1:raise SpeedtestCLIError('--csv-delimiter must be a single character')validate_optional_args(args)socket.setdefaulttimeout(args.timeout)# If specified bind to a specific IP addressif args.source:SOURCE = args.sourcesocket.socket = bound_socketif args.secure:SCHEME = 'https'debug = getattr(args, 'debug', False)if debug == 'SUPPRESSHELP':debug = Falseif debug:DEBUG = True# Pre-cache the user agent stringbuild_user_agent()if args.simple or args.csv or args.json:quiet = Trueelse:quiet = Falseif args.csv or args.json:machine_format = Trueelse:machine_format = False# Don't set a callback if we are running quietlyif quiet or debug:callback = do_nothingelse:callback = print_dotsprinter('\033[00;34m►\033[00;36m=====================SPEED TEST BY \033[00;32mCODEX\033[00;36m=========================\033[00;34m◄ ', quiet)try:speedtest = Speedtest()except (ConfigRetrievalError, HTTP_ERRORS):printer('Cannot retrieve speedtest configuration')raise SpeedtestCLIError(get_exception())if args.list:try:speedtest.get_servers()except (ServersRetrievalError, HTTP_ERRORS):print_('Cannot retrieve speedtest server list')raise SpeedtestCLIError(get_exception())for _, servers in sorted(speedtest.servers.items()):for server in servers:line = ('%(id)5s) %(sponsor)s (%(name)s, %(country)s) ''[%(d)0.2f km]' % server)try:print_(line)except IOError:e = get_exception()if e.errno != errno.EPIPE:raisesys.exit(0)# Set a filter of servers to retrieveservers = []if args.server:servers.append(args.server)printer('\033[01;34m[\033[01;31m₪\033[01;34m]\033[01;35m► \033[00;96mFirma : \033[01;33m%(isp)s \033[01;31m( Ascund IP-ul )... \033[00;31m\033[00;31m\033[01;32m' % speedtest.config['client'],quiet)if not args.mini:
# printer('', quiet)try:speedtest.get_servers(servers)except NoMatchedServers:raise SpeedtestCLIError('No matched servers: %s' % args.server)except (ServersRetrievalError, HTTP_ERRORS):print_('Cannot retrieve speedtest server list')raise SpeedtestCLIError(get_exception())except InvalidServerIDType:raise SpeedtestCLIError('%s is an invalid server type, must ''be an int' % args.server)printer('\033[01;34m[\033[01;31m₪\033[01;34m]\033[01;35m► \033[00;96mGasesc Server \033[01;31m\033[00;31m\033[01;32m', quiet)speedtest.get_best_server()elif args.mini:speedtest.get_best_server(speedtest.set_mini_server(args.mini))results = speedtest.resultsprinter('\033[01;34m[\033[01;31m₪\033[01;34m\033[01;35m]► \033[00;96mHostat de : \033[00;31m%(sponsor)s \033[01;33m(%(name)s) \033[00;32m[%(d)0.2f km]\033[01;35m\033[01;34m\033[01;38m ''%(latency)s ms' % results.server, quiet)if args.download:printer('\033[01;34m[\033[01;31m₪\033[01;34m]\033[01;35m► \033[00;96mIncerc viteza Download \033[01;32m\033[00;31m\033[01;32m', quiet,end=('', '\n')[bool(debug)])speedtest.download(callback=callback)printer('\033[01;34m[\033[01;31m₪\033[01;34m]\033[01;35m► \033[00;96mDOWNLOAD :\033[01;34m %0.2f M%s/s \033[01;32m\033[00;31m\033[01;32m' %((results.download / 1000.0 / 1000.0) / args.units[1],args.units[0]),quiet)else:printer('Skipping download test')if args.upload:printer('\033[01;34m[\033[01;31m₪\033[01;34m]\033[01;35m► \033[00;96mIncerc viteza Upload \033[01;32m\033[00;31m\033[01;32m', quiet,end=('', '\n')[bool(debug)])speedtest.upload(callback=callback, pre_allocate=args.pre_allocate)printer('\033[01;34m[\033[01;31m₪\033[01;34m]\033[01;35m► \033[00;96mUPLOAD :\033[01;34m %0.2f M%s/s \033[01;32m\033[00;31m\033[00;31m' %((results.upload / 1000.0 / 1000.0) / args.units[1],args.units[0]),quiet)else:printer('Skipping upload test')if args.simple:print_('Ping: %s ms\n► Download: %0.2f M%s/s ► \n► Upload: %0.2f M%s/s ► ' %(results.ping,(results.download / 1000.0 / 1000.0) / args.units[1],args.units[0],(results.upload / 1000.0 / 1000.0) / args.units[1],args.units[0]))elif args.csv:print_(results.csv(delimiter=args.csv_delimiter))elif args.json:if args.share:results.share()print_(results.json())if args.share and not machine_format:printer('Share results: %s' % results.share())def main():try:shell()except KeyboardInterrupt:print_('\nAnulăm...')except (SpeedtestException, SystemExit):e = get_exception()if getattr(e, 'code', 1) != 0:raise SystemExit('ERROR: %s' % e)if __name__ == '__main__':main()
这篇关于腾讯云被当做肉鸡了的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!