Vladimir Iakovlev
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
# Initialize output before importing any module, that can use colorama.
|
|
from .system import init_output
|
|
|
|
init_output()
|
|
|
|
import os # noqa: E402
|
|
from psutil import Process # noqa: E402
|
|
import six # noqa: E402
|
|
from . import logs # noqa: E402
|
|
from .shells import shell # noqa: E402
|
|
from .conf import settings # noqa: E402
|
|
from .system import Path # noqa: E402
|
|
from .utils import get_cache_dir # noqa: E402
|
|
|
|
|
|
def _get_shell_pid():
|
|
"""Returns parent process pid."""
|
|
proc = Process(os.getpid())
|
|
|
|
try:
|
|
return proc.parent().pid
|
|
except TypeError:
|
|
return proc.parent.pid
|
|
|
|
|
|
def _get_not_configured_usage_tracker_path():
|
|
"""Returns path of special file where we store latest shell pid."""
|
|
return Path(get_cache_dir()).joinpath('thefuck.last_not_configured_run')
|
|
|
|
|
|
def _record_first_run():
|
|
"""Records shell pid to tracker file."""
|
|
with _get_not_configured_usage_tracker_path().open('w') as tracker:
|
|
tracker.write(six.text_type(_get_shell_pid()))
|
|
|
|
|
|
def _is_second_run():
|
|
"""Returns `True` when we know that `fuck` called second time."""
|
|
tracker_path = _get_not_configured_usage_tracker_path()
|
|
if not tracker_path.exists() or not shell.get_history()[-1] == 'fuck':
|
|
return False
|
|
|
|
current_pid = _get_shell_pid()
|
|
with tracker_path.open('r') as tracker:
|
|
return tracker.read() == six.text_type(current_pid)
|
|
|
|
|
|
def _is_already_configured(configuration_details):
|
|
"""Returns `True` when alias already in shell config."""
|
|
path = Path(configuration_details.path).expanduser()
|
|
with path.open('r') as shell_config:
|
|
return configuration_details.content in shell_config.read()
|
|
|
|
|
|
def _configure(configuration_details):
|
|
"""Adds alias to shell config."""
|
|
path = Path(configuration_details.path).expanduser()
|
|
with path.open('a') as shell_config:
|
|
shell_config.write(u'\n')
|
|
shell_config.write(configuration_details.content)
|
|
shell_config.write(u'\n')
|
|
|
|
|
|
def main():
|
|
"""Shows useful information about how-to configure alias on a first run
|
|
and configure automatically on a second.
|
|
|
|
It'll be only visible when user type fuck and when alias isn't configured.
|
|
|
|
"""
|
|
settings.init()
|
|
configuration_details = shell.how_to_configure()
|
|
if (
|
|
configuration_details and
|
|
configuration_details.can_configure_automatically
|
|
):
|
|
if _is_already_configured(configuration_details):
|
|
logs.already_configured(configuration_details)
|
|
return
|
|
elif _is_second_run():
|
|
_configure(configuration_details)
|
|
logs.configured_successfully(configuration_details)
|
|
return
|
|
else:
|
|
_record_first_run()
|
|
|
|
logs.how_to_configure_alias(configuration_details)
|