mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2025-01-11 07:23:04 +00:00
More Python 3 tweaks.
This commit is contained in:
parent
d84d366c72
commit
90188ce4d1
@ -6,7 +6,7 @@ import sys
|
|||||||
from datetime import (
|
from datetime import (
|
||||||
datetime,
|
datetime,
|
||||||
)
|
)
|
||||||
from typing import Optional
|
from typing import Optional, Union
|
||||||
|
|
||||||
from twisted.python.filepath import FilePath
|
from twisted.python.filepath import FilePath
|
||||||
|
|
||||||
@ -338,7 +338,7 @@ def save_grid_manager(file_path, grid_manager, create=True):
|
|||||||
f.write(b"\n")
|
f.write(b"\n")
|
||||||
|
|
||||||
|
|
||||||
def parse_grid_manager_certificate(gm_data):
|
def parse_grid_manager_certificate(gm_data: Union[str, bytes]):
|
||||||
"""
|
"""
|
||||||
:param gm_data: some data that might be JSON that might be a valid
|
:param gm_data: some data that might be JSON that might be a valid
|
||||||
Grid Manager Certificate
|
Grid Manager Certificate
|
||||||
|
@ -10,12 +10,9 @@ from future.utils import PY2
|
|||||||
if PY2:
|
if PY2:
|
||||||
from future.builtins import filter, map, zip, ascii, chr, hex, input, next, oct, open, pow, round, super, bytes, dict, list, object, range, str, max, min # noqa: F401
|
from future.builtins import filter, map, zip, ascii, chr, hex, input, next, oct, open, pow, round, super, bytes, dict, list, object, range, str, max, min # noqa: F401
|
||||||
|
|
||||||
from six import ensure_binary, ensure_str
|
from six import ensure_binary
|
||||||
|
|
||||||
try:
|
from typing import Union
|
||||||
from allmydata.scripts.types_ import SubCommands
|
|
||||||
except ImportError:
|
|
||||||
pass
|
|
||||||
|
|
||||||
from twisted.python import usage
|
from twisted.python import usage
|
||||||
from twisted.python.filepath import (
|
from twisted.python.filepath import (
|
||||||
@ -29,14 +26,12 @@ from allmydata.storage import (
|
|||||||
crawler,
|
crawler,
|
||||||
expirer,
|
expirer,
|
||||||
)
|
)
|
||||||
from twisted.python.filepath import FilePath
|
from allmydata.scripts.types_ import SubCommands
|
||||||
|
|
||||||
from allmydata.client import read_config
|
from allmydata.client import read_config
|
||||||
from allmydata.grid_manager import (
|
from allmydata.grid_manager import (
|
||||||
parse_grid_manager_certificate,
|
parse_grid_manager_certificate,
|
||||||
)
|
)
|
||||||
from allmydata.scripts.cli import _default_nodedir
|
from allmydata.scripts.cli import _default_nodedir
|
||||||
from allmydata.scripts.common import BaseOptions
|
|
||||||
from allmydata.util.encodingutil import argv_to_abspath
|
from allmydata.util.encodingutil import argv_to_abspath
|
||||||
from allmydata.util import jsonbytes
|
from allmydata.util import jsonbytes
|
||||||
|
|
||||||
@ -127,8 +122,10 @@ class AddGridManagerCertOptions(BaseOptions):
|
|||||||
raise usage.UsageError(
|
raise usage.UsageError(
|
||||||
"Must provide --filename option"
|
"Must provide --filename option"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
data : Union [bytes, str]
|
||||||
if self['filename'] == '-':
|
if self['filename'] == '-':
|
||||||
print(ensure_str("reading certificate from stdin"), file=self.parent.parent.stderr)
|
print("reading certificate from stdin", file=self.parent.parent.stderr)
|
||||||
data = self.parent.parent.stdin.read()
|
data = self.parent.parent.stdin.read()
|
||||||
if len(data) == 0:
|
if len(data) == 0:
|
||||||
raise usage.UsageError(
|
raise usage.UsageError(
|
||||||
@ -201,7 +198,7 @@ def add_grid_manager_cert(options):
|
|||||||
options['name'],
|
options['name'],
|
||||||
cert_path.path,
|
cert_path.path,
|
||||||
)
|
)
|
||||||
print(ensure_str(msg), file=options.parent.parent.stderr)
|
print(msg, file=options.parent.parent.stderr)
|
||||||
return 1
|
return 1
|
||||||
|
|
||||||
config.set_config("storage", "grid_management", "True")
|
config.set_config("storage", "grid_management", "True")
|
||||||
@ -212,7 +209,7 @@ def add_grid_manager_cert(options):
|
|||||||
f.write(cert_bytes)
|
f.write(cert_bytes)
|
||||||
|
|
||||||
cert_count = len(config.enumerate_section("grid_manager_certificates"))
|
cert_count = len(config.enumerate_section("grid_manager_certificates"))
|
||||||
print(ensure_str("There are now {} certificates").format(cert_count),
|
print("There are now {} certificates".format(cert_count),
|
||||||
file=options.parent.parent.stderr)
|
file=options.parent.parent.stderr)
|
||||||
|
|
||||||
return 0
|
return 0
|
||||||
|
@ -2,8 +2,6 @@
|
|||||||
Type definitions used by modules in this package.
|
Type definitions used by modules in this package.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# Python 3 only
|
|
||||||
|
|
||||||
from typing import List, Tuple, Type, Sequence, Any
|
from typing import List, Tuple, Type, Sequence, Any
|
||||||
from twisted.python.usage import Options
|
from twisted.python.usage import Options
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user