client.py

#!/usr/bin/env python

import csv
import datetime
import platform as plf
import re
import subprocess
import timeit


OPENBSD_CPU_SAMPLE_LINE = (
    "cpu0: Intel(R) Core(TM) i5-8265U CPU @ 1.60GHz, 15430.67 MHz, 06-8e-0c"
)
OPENBSD_CPU_SAMPLE_MODEL = "Intel(R) Core(TM) i5-8265U CPU @ 1.60GHz"
GITHUB_ISSUE = "https://github.com/pymivn/cpuisfast/issues/new"


def get_processor_name():
    # type: () -> str
    if plf.system() == "Windows":
        try:
            import winreg

            key = winreg.OpenKey(
                winreg.HKEY_LOCAL_MACHINE,
                r"Hardware\Description\System\CentralProcessor\0",
            )  # NOQA
            processor_brand = winreg.QueryValueEx(key, "ProcessorNameString")[
                0
            ]
            winreg.CloseKey(key)
            return processor_brand
        except Exception:
            return plf.processor()
    elif plf.system() == "Darwin":
        command = ["sysctl", "-n", "machdep.cpu.brand_string"]
        return subprocess.check_output(command).strip().decode()
    elif plf.system() == "Linux":
        with open("/proc/cpuinfo") as f:
            for line in f:
                if "model name" in line:
                    return re.sub(".*model name.*:", "", line, 1)
    elif "bsd" in plf.system().lower():

        def parse_openbsd_cpu(line):
            m = re.match("^cpu[0-9]+:", line, re.IGNORECASE)
            if m is None:
                return m

            return line.replace(m.group(), "").strip().split(",")[0]

        assert (
            parse_openbsd_cpu(OPENBSD_CPU_SAMPLE_LINE)
            == OPENBSD_CPU_SAMPLE_MODEL
        )

        try:
            with open("/var/run/dmesg.boot") as f:
                for line in f:
                    r = parse_openbsd_cpu(line)
                    if r:
                        return r
        except Exception as e:
            print("Cannot get *BSD CPU %s", e)

    return "" or plf.machine()


ONE_MILLION = 10 ** 6
t_empty = timeit.Timer(
    "for i in range({}):\n    pass\n".format(ONE_MILLION)
)  # NOQA
t = timeit.Timer(
    "s = 0\nfor i in range({}):\n    s = s + i\nprint(s)".format(ONE_MILLION)
)  # NOQA

# read timeit doc for why use min
took_empty = t_empty.repeat(10, 1)
empty_best = min(took_empty)

took = t.repeat(10, 1)
best = min(took)
cpu_name = get_processor_name()
empty_iterate_freq = ONE_MILLION // empty_best
plus_iterate_freq = ONE_MILLION // best
print("Your CPU is %s" % cpu_name)
arch = plf.machine()
implementation = plf.python_implementation()
py_ver = plf.python_version()
system = plf.system()
print("This is running on", system, arch, implementation, py_ver)


print("It can loop %d times per second" % empty_iterate_freq)
print("Calculating sum from 0 to %d" % ONE_MILLION)
print("Best took %s s" % best)
print("Your CPU can do: %d (+) operations/second" % plus_iterate_freq)
data = {
    "cpu": cpu_name.strip(),
    "empty_iterate_freq": empty_iterate_freq,
    "plus_iterate_freq": plus_iterate_freq,
    "system": system,
    "arch": arch,
    "implementation": implementation,
    "python_version": py_ver,
    "inserted_at": datetime.datetime.now().isoformat().replace('T', ' '),
}

with open("cpuisfast/data.csv", 'at') as f:
    writer = csv.writer(f)
    writer.writerow(data.values())

print("Done, please push then open new Pull Request to submit your result.")