sapling/eden/scm/edenscm/__init__.py
Jun Wu 3e0b781197 py3: only use binary stdin/stdout/stderr
Summary:
Drop stdoutbytes/stdinbytes. They make things unnecessarily complicated
(especially for chg / Rust dispatch entry point).

The new idea is IO are using bytes. Text are written in utf-8 (Python 3) or
local encoding (Python 2). To make stdout behave reasonably on systems not
using utf-8 locale (ex. Windows), we might add a Rust binding to Rust's stdout,
which does the right thing:
- When writing to stdout console, expect text to be utf-8 encoded and do proper decoding.
- Wehn writing to stdout file, write the raw bytes without translation.

Note Python's `sys.stdout.buffer` does not do translation when writing to stdout console
like Rust's stdout.

For now, my main motivation of this change is to fix chg on Python 3.

Reviewed By: xavierd

Differential Revision: D19702533

fbshipit-source-id: 74704c83e1b200ff66fb3a2d23d97ff21c7239c8
2020-02-03 18:26:57 -08:00

87 lines
2.6 KiB
Python

# Copyright (c) Facebook, Inc. and its affiliates.
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2.
from __future__ import absolute_import
def _fixsyspath():
"""Fix sys.path so core edenscm modules (edenscmnative, and 3rd party
libraries) are in sys.path
"""
# Do not expose those modules to edenscm.__dict__
import sys
import os
dirname = os.path.dirname
# __file__ is "hg/edenscm/__init__.py"
# libdir is "hg/"
# Do not follow symlinks (ex. do not use "realpath"). It breaks buck build.
libdir = dirname(dirname(os.path.abspath(__file__)))
# Make "edenscmdeps.zip" available in sys.path. It includes 3rd party
# pure-Python libraries like IPython, thrift runtime, etc.
#
# Note: On Windows, the released version of hg uses python27.zip for all
# pure Python modules including edenscm and everything in edenscmdeps.zip,
# so not being able to locate edenscmdeps.zip is not fatal.
if sys.version_info[0] == 3:
name = "edenscmdeps3.zip"
else:
name = "edenscmdeps.zip"
for candidate in [libdir, os.path.join(libdir, "build")]:
depspath = os.path.join(candidate, name)
if os.path.exists(depspath) and depspath not in sys.path:
sys.path.insert(0, depspath)
# Make sure "edenscmnative" can be imported. Error early.
# pyre-fixme[21]: Could not find `edenscmnative`.
import edenscmnative
edenscmnative.__name__
_fixsyspath()
# Keep the module clean
del globals()["_fixsyspath"]
def run(args=None, fin=None, fout=None, ferr=None):
import sys
if args is None:
args = sys.argv
if args[1:4] == ["serve", "--cmdserver", "chgunix2"]:
# chgserver code path
# no demandimport, since chgserver wants to preimport everything.
from .mercurial import dispatch
dispatch.runchgserver()
else:
# non-chgserver code path
# - no chg in use: hgcommands::run -> HgPython::run_hg -> here
# - chg client: chgserver.runcommand -> bindings.commands.run ->
# hgcommands::run -> HgPython::run_hg -> here
from . import traceimport
traceimport.enable()
if sys.version_info[0] < 3:
# enable demandimport after enabling traceimport
from . import hgdemandimport
hgdemandimport.enable()
# demandimport has side effect on importing dispatch.
# so 'import dispatch' happens after demandimport
from .mercurial import dispatch
dispatch.run(args, fin, fout, ferr)