forked from platformio/platformio-core
100 lines
2.4 KiB
Python
100 lines
2.4 KiB
Python
# Copyright (C) Ivan Kravets <me@ikravets.com>
|
|
# See LICENSE for details.
|
|
|
|
from os import name as os_name
|
|
from os import getcwd, getenv, listdir, utime
|
|
from os.path import dirname, expanduser, isfile, join, realpath
|
|
from platform import system, uname
|
|
from subprocess import PIPE, Popen
|
|
from time import sleep
|
|
|
|
from serial import Serial
|
|
|
|
from platformio.exception import GetSerialPortsError, NotPlatformProject
|
|
|
|
try:
|
|
from configparser import ConfigParser
|
|
except ImportError:
|
|
from ConfigParser import ConfigParser
|
|
|
|
|
|
def get_systype():
|
|
if system() == "Windows":
|
|
return "windows"
|
|
data = uname()
|
|
return ("%s_%s" % (data[0], data[4])).lower()
|
|
|
|
|
|
def get_home_dir():
|
|
home_dir = expanduser("~/.platformio")
|
|
try:
|
|
config = get_project_config()
|
|
if (config.has_section("platformio") and
|
|
config.has_option("platformio", "home_dir")):
|
|
return config.get("platformio", "home_dir")
|
|
except NotPlatformProject:
|
|
pass
|
|
return home_dir
|
|
|
|
|
|
def get_source_dir():
|
|
return dirname(realpath(__file__))
|
|
|
|
|
|
def get_project_dir():
|
|
return getcwd()
|
|
|
|
|
|
def get_pioenvs_dir():
|
|
return getenv("PIOENVS_DIR", join(get_project_dir(), ".pioenvs"))
|
|
|
|
|
|
def get_project_config():
|
|
path = join(get_project_dir(), "platformio.ini")
|
|
if not isfile(path):
|
|
raise NotPlatformProject()
|
|
cp = ConfigParser()
|
|
cp.read(path)
|
|
return cp
|
|
|
|
|
|
def get_platforms():
|
|
platforms = []
|
|
for p in listdir(join(get_source_dir(), "platforms")):
|
|
if p in ("__init__.py", "base.py") or not p.endswith(".py"):
|
|
continue
|
|
platforms.append(p[:-3])
|
|
return platforms
|
|
|
|
|
|
def change_filemtime(path, time):
|
|
utime(path, (time, time))
|
|
|
|
|
|
def exec_command(args):
|
|
use_shell = system() == "Windows"
|
|
p = Popen(args, stdout=PIPE, stderr=PIPE, shell=use_shell)
|
|
out, err = p.communicate()
|
|
return dict(out=out.strip(), err=err.strip())
|
|
|
|
|
|
def reset_serialport(port):
|
|
s = Serial(port)
|
|
s.flushInput()
|
|
s.setDTR(False)
|
|
s.setRTS(False)
|
|
sleep(0.1)
|
|
s.setDTR(True)
|
|
s.setRTS(True)
|
|
s.close()
|
|
|
|
|
|
def get_serialports():
|
|
if os_name == "nt":
|
|
from serial.tools.list_ports_windows import comports
|
|
elif os_name == "posix":
|
|
from serial.tools.list_ports_posix import comports
|
|
else:
|
|
raise GetSerialPortsError(os_name)
|
|
return[{"port": p, "description": d, "hwid": h} for p, d, h in comports()]
|