forked from extern/httpie-cli
129 lines
3.9 KiB
Python
129 lines
3.9 KiB
Python
import sys
|
||
import os
|
||
from pathlib import Path
|
||
from typing import IO, Optional
|
||
|
||
|
||
try:
|
||
import curses
|
||
except ImportError:
|
||
curses = None # Compiled w/o curses
|
||
|
||
from httpie.compat import is_windows
|
||
from httpie.config import DEFAULT_CONFIG_DIR, Config, ConfigFileError
|
||
|
||
from httpie.utils import repr_dict
|
||
|
||
|
||
class Environment:
|
||
"""
|
||
Information about the execution context
|
||
(standard streams, config directory, etc).
|
||
|
||
By default, it represents the actual environment.
|
||
All of the attributes can be overwritten though, which
|
||
is used by the test suite to simulate various scenarios.
|
||
|
||
"""
|
||
is_windows: bool = is_windows
|
||
config_dir: Path = DEFAULT_CONFIG_DIR
|
||
stdin: Optional[IO] = sys.stdin # `None` when closed fd (#791)
|
||
stdin_isatty: bool = stdin.isatty() if stdin else False
|
||
stdin_encoding: str = None
|
||
stdout: IO = sys.stdout
|
||
stdout_isatty: bool = stdout.isatty()
|
||
stdout_encoding: str = None
|
||
stderr: IO = sys.stderr
|
||
stderr_isatty: bool = stderr.isatty()
|
||
colors = 256
|
||
program_name: str = 'http'
|
||
if not is_windows:
|
||
if curses:
|
||
try:
|
||
curses.setupterm()
|
||
colors = curses.tigetnum('colors')
|
||
except curses.error:
|
||
pass
|
||
else:
|
||
# noinspection PyUnresolvedReferences
|
||
import colorama.initialise
|
||
stdout = colorama.initialise.wrap_stream(
|
||
stdout, convert=None, strip=None,
|
||
autoreset=True, wrap=True
|
||
)
|
||
stderr = colorama.initialise.wrap_stream(
|
||
stderr, convert=None, strip=None,
|
||
autoreset=True, wrap=True
|
||
)
|
||
del colorama
|
||
|
||
def __init__(self, devnull=None, **kwargs):
|
||
"""
|
||
Use keyword arguments to overwrite
|
||
any of the class attributes for this instance.
|
||
|
||
"""
|
||
assert all(hasattr(type(self), attr) for attr in kwargs.keys())
|
||
self.__dict__.update(**kwargs)
|
||
|
||
# The original STDERR unaffected by --quiet’ing.
|
||
self._orig_stderr = self.stderr
|
||
self._devnull = devnull
|
||
|
||
# Keyword arguments > stream.encoding > default utf8
|
||
if self.stdin and self.stdin_encoding is None:
|
||
self.stdin_encoding = getattr(
|
||
self.stdin, 'encoding', None) or 'utf8'
|
||
if self.stdout_encoding is None:
|
||
actual_stdout = self.stdout
|
||
if is_windows:
|
||
# noinspection PyUnresolvedReferences
|
||
from colorama import AnsiToWin32
|
||
if isinstance(self.stdout, AnsiToWin32):
|
||
# noinspection PyUnresolvedReferences
|
||
actual_stdout = self.stdout.wrapped
|
||
self.stdout_encoding = getattr(
|
||
actual_stdout, 'encoding', None) or 'utf8'
|
||
|
||
def __str__(self):
|
||
defaults = dict(type(self).__dict__)
|
||
actual = dict(defaults)
|
||
actual.update(self.__dict__)
|
||
actual['config'] = self.config
|
||
return repr_dict({
|
||
key: value
|
||
for key, value in actual.items()
|
||
if not key.startswith('_')
|
||
})
|
||
|
||
def __repr__(self):
|
||
return f'<{type(self).__name__} {self}>'
|
||
|
||
_config: Config = None
|
||
|
||
@property
|
||
def config(self) -> Config:
|
||
config = self._config
|
||
if not config:
|
||
self._config = config = Config(directory=self.config_dir)
|
||
if not config.is_new():
|
||
try:
|
||
config.load()
|
||
except ConfigFileError as e:
|
||
self.log_error(e, level='warning')
|
||
return config
|
||
|
||
@property
|
||
def devnull(self) -> IO:
|
||
if self._devnull is None:
|
||
self._devnull = open(os.devnull, 'w+')
|
||
return self._devnull
|
||
|
||
@devnull.setter
|
||
def devnull(self, value):
|
||
self._devnull = value
|
||
|
||
def log_error(self, msg, level='error'):
|
||
assert level in ['error', 'warning']
|
||
self._orig_stderr.write(f'\n{self.program_name}: {level}: {msg}\n\n')
|