Current File : //lib/python3/dist-packages/twisted/words/tap.py
# -*- test-case-name: twisted.words.test.test_tap -*-
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.
"""
Shiny new words service maker
"""
import socket
import sys
from typing import List, Optional, Sequence
from twisted import plugin
from twisted.application import strports
from twisted.application.service import MultiService
from twisted.cred import checkers, credentials, portal, strcred
from twisted.python import usage
from twisted.words import iwords, service
class Options(usage.Options, strcred.AuthOptionMixin):
supportedInterfaces = [credentials.IUsernamePassword]
optParameters: List[Sequence[Optional[str]]] = [
(
"hostname",
None,
socket.gethostname(),
"Name of this server; purely an informative",
)
]
compData = usage.Completions(multiUse=["group"])
interfacePlugins = {}
plg = None
for plg in plugin.getPlugins(iwords.IProtocolPlugin):
assert plg.name not in interfacePlugins
interfacePlugins[plg.name] = plg
optParameters.append(
(
plg.name + "-port",
None,
None,
"strports description of the port to bind for the "
+ plg.name
+ " server",
)
)
del plg
def __init__(self, *a, **kw):
usage.Options.__init__(self, *a, **kw)
self["groups"] = []
def opt_group(self, name):
"""Specify a group which should exist"""
self["groups"].append(name.decode(sys.stdin.encoding))
def opt_passwd(self, filename):
"""
Name of a passwd-style file. (This is for
backwards-compatibility only; you should use the --auth
command instead.)
"""
self.addChecker(checkers.FilePasswordDB(filename))
def makeService(config):
credCheckers = config.get("credCheckers", [])
wordsRealm = service.InMemoryWordsRealm(config["hostname"])
wordsPortal = portal.Portal(wordsRealm, credCheckers)
msvc = MultiService()
# XXX Attribute lookup on config is kind of bad - hrm.
for plgName in config.interfacePlugins:
port = config.get(plgName + "-port")
if port is not None:
factory = config.interfacePlugins[plgName].getFactory(
wordsRealm, wordsPortal
)
svc = strports.service(port, factory)
svc.setServiceParent(msvc)
# This is bogus. createGroup is async. makeService must be
# allowed to return a Deferred or some crap.
for g in config["groups"]:
wordsRealm.createGroup(g)
return msvc
Mini Shell