Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enhanced code readability #62

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 7 additions & 5 deletions kubeshell/client.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,13 @@
from __future__ import absolute_import, unicode_literals, print_function
from urllib3.exceptions import NewConnectionError, ConnectTimeoutError, MaxRetryError
from kubernetes import client, config
from kubernetes.client.api_client import ApiException
from __future__ import absolute_import, print_function, unicode_literals

import os
import logging
import os

import urllib3
from kubernetes import client, config
from kubernetes.client.api_client import ApiException
from urllib3.exceptions import (ConnectTimeoutError, MaxRetryError,
NewConnectionError)

# disable warnings on stdout/stderr from urllib3 connection errors
ulogger = logging.getLogger("urllib3")
Expand Down
16 changes: 9 additions & 7 deletions kubeshell/completer.py
Original file line number Diff line number Diff line change
@@ -1,15 +1,17 @@
from __future__ import absolute_import, unicode_literals, print_function
from subprocess import check_output
from prompt_toolkit.completion import Completer, Completion
from fuzzyfinder import fuzzyfinder
import logging
import shlex
from __future__ import absolute_import, print_function, unicode_literals

import json
import logging
import os
import os.path
import shlex
from subprocess import check_output

from kubeshell.parser import Parser
from fuzzyfinder import fuzzyfinder
from kubeshell.client import KubernetesClient
from kubeshell.parser import Parser
from prompt_toolkit.completion import Completer, Completion

logger = logging.getLogger(__name__)


Expand Down
28 changes: 14 additions & 14 deletions kubeshell/kubeshell.py
Original file line number Diff line number Diff line change
@@ -1,23 +1,23 @@
from __future__ import print_function, absolute_import, unicode_literals
from __future__ import absolute_import, print_function, unicode_literals

from prompt_toolkit import prompt
from prompt_toolkit.history import FileHistory
from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
from prompt_toolkit.key_binding.defaults import load_key_bindings_for_prompt
from prompt_toolkit.keys import Keys
import logging
import os
import subprocess
import sys

from kubeshell.style import StyleFactory
import click
import yaml
from kubeshell.client import KubernetesClient, kubeconfig_filepath
from kubeshell.completer import KubectlCompleter
from kubeshell.lexer import KubectlLexer
from kubeshell.style import StyleFactory
from kubeshell.toolbar import Toolbar
from kubeshell.client import KubernetesClient, kubeconfig_filepath
from prompt_toolkit import prompt
from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
from prompt_toolkit.history import FileHistory
from prompt_toolkit.key_binding.defaults import load_key_bindings_for_prompt
from prompt_toolkit.keys import Keys

import os
import click
import sys
import subprocess
import yaml
import logging
logger = logging.getLogger(__name__)

inline_help = True
Expand Down
2 changes: 1 addition & 1 deletion kubeshell/logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,4 +50,4 @@
}
},
}
logging.config.dictConfig(loggingConf)
logging.config.dictConfig(loggingConf)
9 changes: 6 additions & 3 deletions kubeshell/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,17 +2,20 @@
# -*- coding: utf-8 -*-


from __future__ import print_function, absolute_import, unicode_literals
from kubeshell.kubeshell import Kubeshell
from __future__ import absolute_import, print_function, unicode_literals

import logging

from kubeshell.kubeshell import Kubeshell

logger = logging.getLogger(__name__)


def cli():
kube_shell= Kubeshell()
kube_shell = Kubeshell()
logger.info("session start")
kube_shell.run_cli()


if __name__ == "__main__":
cli()
9 changes: 5 additions & 4 deletions kubeshell/parser.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
from __future__ import absolute_import, unicode_literals, print_function
import json
import os
from __future__ import absolute_import, print_function, unicode_literals

import json
import logging
logger = logging.getLogger(__name__)

from kubeshell.client import KubernetesClient

logger = logging.getLogger(__name__)


class Option(object):
""" Option represents an optional local flag in kubectl """
Expand Down Expand Up @@ -164,6 +164,7 @@ def evalOptions(self, root, parsed, unparsed):
unparsed.append(token)
return parsed, unparsed, suggestions


if __name__ == '__main__':
parser = Parser('/Users/tsp/workspace/py/kube-shell/kubeshell/data/cli.json')
p, _, s = parser.treewalk(parser.ast, parsed=list(), unparsed=['--', '--tcp 900:8080', 'nodeport', 'service', 'create', 'kubectl'])
Expand Down
7 changes: 4 additions & 3 deletions kubeshell/style.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,11 +10,12 @@
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
# ANY KIND, either express or implied. See the License for the specific
# language governing permissions and limitations under the License.
from __future__ import print_function, absolute_import, unicode_literals
from __future__ import absolute_import, print_function, unicode_literals

from prompt_toolkit.styles import default_style_extensions, style_from_dict
from pygments.styles import get_style_by_name
from pygments.token import Token
from pygments.util import ClassNotFound
from pygments.styles import get_style_by_name
from prompt_toolkit.styles import default_style_extensions, style_from_dict


class StyleFactory(object):
Expand Down
6 changes: 4 additions & 2 deletions kubeshell/tests/test_cli.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
from __future__ import unicode_literals

import unittest
import pip

import pexpect
import unittest


class CliTest(unittest.TestCase):

Expand All @@ -17,5 +18,6 @@ def step_run_cli(self):
def step_see_prompt(self):
self.cli.expect('kube-shell> ')


if __name__ == "__main__":
unittest.main()
8 changes: 5 additions & 3 deletions kubeshell/toolbar.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
from __future__ import print_function, absolute_import, unicode_literals
from pygments.token import Token
from pygments.token import Keyword, Name, Operator, Generic, Literal, Text
from __future__ import absolute_import, print_function, unicode_literals

from pygments.token import (Generic, Keyword, Literal, Name, Operator, Text,
Token)


class Toolbar(object):
"""Show information about the aws-shell in a tool bar.
Expand Down