89 lines
2.7 KiB
Python
89 lines
2.7 KiB
Python
# Low level unix utility functions
|
|
#
|
|
# Copyright (C) 2016 Kevin O'Connor <kevin@koconnor.net>
|
|
#
|
|
# This file may be distributed under the terms of the GNU GPLv3 license.
|
|
import sys, os, pty, fcntl, termios, signal, logging
|
|
import subprocess, traceback, shlex
|
|
|
|
# Return the SIGINT interrupt handler back to the OS default
|
|
def fix_sigint():
|
|
signal.signal(signal.SIGINT, signal.SIG_DFL)
|
|
fix_sigint()
|
|
|
|
# Set a file-descriptor as non-blocking
|
|
def set_nonblock(fd):
|
|
fcntl.fcntl(fd, fcntl.F_SETFL
|
|
, fcntl.fcntl(fd, fcntl.F_GETFL) | os.O_NONBLOCK)
|
|
|
|
# Clear HUPCL flag
|
|
def clear_hupcl(fd):
|
|
attrs = termios.tcgetattr(fd)
|
|
attrs[2] = attrs[2] & ~termios.HUPCL
|
|
try:
|
|
termios.tcsetattr(fd, termios.TCSADRAIN, attrs)
|
|
except termios.error:
|
|
pass
|
|
|
|
# Support for creating a pseudo-tty for emulating a serial port
|
|
def create_pty(ptyname):
|
|
mfd, sfd = pty.openpty()
|
|
try:
|
|
os.unlink(ptyname)
|
|
except os.error:
|
|
pass
|
|
filename = os.ttyname(sfd)
|
|
os.chmod(filename, 0660)
|
|
os.symlink(filename, ptyname)
|
|
set_nonblock(mfd)
|
|
old = termios.tcgetattr(mfd)
|
|
old[3] = old[3] & ~termios.ECHO
|
|
termios.tcsetattr(mfd, termios.TCSADRAIN, old)
|
|
return mfd
|
|
|
|
def get_cpu_info():
|
|
try:
|
|
f = open('/proc/cpuinfo', 'rb')
|
|
data = f.read()
|
|
f.close()
|
|
except IOError, OSError:
|
|
logging.debug("Exception on read /proc/cpuinfo: %s",
|
|
traceback.format_exc())
|
|
return "?"
|
|
lines = [l.split(':', 1) for l in data.split('\n')]
|
|
lines = [(l[0].strip(), l[1].strip()) for l in lines if len(l) == 2]
|
|
core_count = [k for k, v in lines].count("processor")
|
|
model_name = dict(lines).get("model name", "?")
|
|
return "%d core %s" % (core_count, model_name)
|
|
|
|
def get_version_from_file(klippy_src):
|
|
try:
|
|
with open(os.path.join(klippy_src, '.version')) as h:
|
|
return h.read().rstrip()
|
|
except IOError:
|
|
pass
|
|
return "?"
|
|
|
|
def get_git_version(from_file=True):
|
|
klippy_src = os.path.dirname(__file__)
|
|
|
|
# Obtain version info from "git" program
|
|
gitdir = os.path.join(klippy_src, '..')
|
|
prog = ('git', '-C', gitdir, 'describe', '--always',
|
|
'--tags', '--long', '--dirty')
|
|
try:
|
|
process = subprocess.Popen(prog, stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE)
|
|
ver, err = process.communicate()
|
|
retcode = process.wait()
|
|
if retcode == 0:
|
|
return ver.strip()
|
|
else:
|
|
logging.debug("Error getting git version: %s", err)
|
|
except OSError:
|
|
logging.debug("Exception on run: %s", traceback.format_exc())
|
|
|
|
if from_file:
|
|
return get_version_from_file(klippy_src)
|
|
return "?"
|