keanucode
/
dotfiles
mirror of https://github.com/keanuplayz/dotfiles.git
1
0
Fork 0
Code Issues 1 Projects Releases Wiki Activity

[welcome] apply flake8's hints

This commit is contained in:
Dmytro Meleshko 2018-08-08 23:08:05 +03:00
parent 3b2f56fa65
commit 190c04cb4b
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
welcome/main.py
View File

@ -1,11 +1,12 @@
import os import os
import platform import platform
import re
import socket import socket
from datetime import datetime, timedelta from datetime import datetime, timedelta
import re
from getpass import getuser from getpass import getuser
from colorama import Fore, Back, Style, ansi
import psutil import psutil
from colorama import Fore, Style, ansi
COLORS = [ansi.code_to_chars(30 + color_index) for color_index in range(0, 8)] COLORS = [ansi.code_to_chars(30 + color_index) for color_index in range(0, 8)]
@ -44,13 +45,13 @@ def humanize_bytes(bytes):
units = ["B", "kB", "MB", "GB"] units = ["B", "kB", "MB", "GB"]
factor = 1 factor = 1
for unit in units: for _unit in units:
next_factor = factor << 10 next_factor = factor << 10
if bytes < next_factor: if bytes < next_factor:
break break
factor = next_factor factor = next_factor
return "%.2f %s" % (float(bytes) / factor, unit) return "%.2f %s" % (float(bytes) / factor, _unit)
def colorize_percent(percent, warning, critical, inverse=False): def colorize_percent(percent, warning, critical, inverse=False):
@ -125,8 +126,9 @@ def disks():
result = [] result = []
for disk in psutil.disk_partitions(all=False): for disk in psutil.disk_partitions(all=False):
if psutil.WINDOWS and ("cdrom" in disk.opts or disk.fstype == ""): if psutil.WINDOWS and ("cdrom" in disk.opts or disk.fstype == ""):
# skip cd-rom drives with no disk in it on Windows; they may raise ENOENT, # skip cd-rom drives with no disk in it on Windows; they may raise
# pop-up a Windows GUI error for a non-ready partition or just hang # ENOENT, pop-up a Windows GUI error for a non-ready partition or
# just hang
continue continue
usage = psutil.disk_usage(disk.mountpoint) usage = psutil.disk_usage(disk.mountpoint)
@ -185,8 +187,8 @@ def get_distro_info():
import distro import distro
return distro.id(), distro.name(), distro.version(), distro.codename() return distro.id(), distro.name(), distro.version(), distro.codename()
else:
raise NotImplementedError("unsupported OS") raise NotImplementedError("unsupported OS")
def get_system_info(): def get_system_info():
@ -268,4 +270,3 @@ for line_index in range(0, max(len(logo_lines), len(info_lines))):
print(line) print(line)
print("") print("")