测试web端登录docker容器

Author: pythonzm

.gitignore

@@ -30,6 +30,7 @@ wheels/
 .installed.cfg
 *.egg
 MANIFEST
+.idea
 
 # PyInstaller
 #  Usually these files are written by a python script from a template

Ops/celery.py

@@ -1,44 +1,44 @@
-# -*- coding: utf-8 -*-
-"""
--------------------------------------------------
-   File Name：      celery
-   Description:
-   Author:          Administrator
-   date：           2018-07-16
--------------------------------------------------
-   Change Activity:
-                    2018-07-16:
--------------------------------------------------
-"""
-import os
-from celery import Celery
-from kombu import Queue, Exchange
-
-# set the default Django settings module for the 'celery' program.
-os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'Ops.settings')
-
-app = Celery('Ops', broker='redis://127.0.0.1:6379/0')
-
-app.conf.task_queues = (
-    Queue('default', Exchange('default', type='direct'), routing_key='default'),
-    Queue('ansible', Exchange('ansible', type='direct'), routing_key='ansible'),
-    Queue('fort', Exchange('fort', type='direct'), routing_key='fort'),
-    Queue('plan', Exchange('plan', type='direct'), routing_key='plan'),
-    Queue('commons', Exchange('commons', type='direct'), routing_key='commons'),
-)
-
-app.conf.task_default_queue = 'default'
-app.conf.task_default_exchange = 'default'
-app.conf.task_default_exchange_type = 'direct'
-app.conf.task_default_routing_key = 'default'
-app.conf.timezone = 'Asia/Shanghai'
-app.conf.enable_utc = False
-
-# Using a string here means the worker doesn't have to serialize
-# the configuration object to child processes.
-# - namespace='CELERY' means all celery-related configuration keys
-#   should have a `CELERY_` prefix.
-app.config_from_object('django.conf:settings', namespace='CELERY')
-
-# Load task modules from all registered Django app configs.
-app.autodiscover_tasks()
+# -*- coding: utf-8 -*-
+"""
+-------------------------------------------------
+   File Name：      celery
+   Description:
+   Author:          Administrator
+   date：           2018-07-16
+-------------------------------------------------
+   Change Activity:
+                    2018-07-16:
+-------------------------------------------------
+"""
+import os
+from celery import Celery
+from kombu import Queue, Exchange
+
+# set the default Django settings module for the 'celery' program.
+os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'Ops.settings')
+
+app = Celery('Ops', broker='redis://127.0.0.1:7779/0')
+
+app.conf.task_queues = (
+    Queue('default', Exchange('default', type='direct'), routing_key='default'),
+    Queue('ansible', Exchange('ansible', type='direct'), routing_key='ansible'),
+    Queue('fort', Exchange('fort', type='direct'), routing_key='fort'),
+    Queue('plan', Exchange('plan', type='direct'), routing_key='plan'),
+    Queue('commons', Exchange('commons', type='direct'), routing_key='commons'),
+)
+
+app.conf.task_default_queue = 'default'
+app.conf.task_default_exchange = 'default'
+app.conf.task_default_exchange_type = 'direct'
+app.conf.task_default_routing_key = 'default'
+app.conf.timezone = 'Asia/Shanghai'
+app.conf.enable_utc = False
+
+# Using a string here means the worker doesn't have to serialize
+# the configuration object to child processes.
+# - namespace='CELERY' means all celery-related configuration keys
+#   should have a `CELERY_` prefix.
+app.config_from_object('django.conf:settings', namespace='CELERY')
+
+# Load task modules from all registered Django app configs.
+app.autodiscover_tasks()

Ops/routing.py

@@ -1,40 +1,42 @@
-# -*- coding: utf-8 -*-
-"""
--------------------------------------------------
-   File Name：      routing
-   Description:
-   Author:          pythonzm
-   date：           2018/6/6
--------------------------------------------------
-   Change Activity:
-                    2018/6/6:
--------------------------------------------------
-"""
-from django.urls import path, re_path
-from channels.routing import ProtocolTypeRouter, URLRouter
-from channels.auth import AuthMiddlewareStack
-from projs.utils.log_websocket import LogConsumer
-from fort.utils.webssh import FortConsumer
-from assets.utils.webssh import SSHConsumer
-from fort.utils.webguacamole import GuacamoleConsumer
-from assets.utils.webguacamole import AdminGuacamole
-from task.utils.ans_module_websocket import AnsModuleConsumer
-from task.utils.ans_playbook_websocket import AnsPlaybookConsumer
-from projs.utils.deploy_websocket import DeployConsumer
-
-application = ProtocolTypeRouter({
-
-    "websocket": AuthMiddlewareStack(
-        URLRouter([
-            # URLRouter just takes standard Django path() or url() entries.
-            path(r'ws/deploy/', DeployConsumer),
-            path(r'ws/ans_module_log/', AnsModuleConsumer),
-            path(r'ws/ans_playbook_log/', AnsPlaybookConsumer),
-            path(r'ws/deploy_log/', LogConsumer),
-            re_path(r'ws/fortssh/([0-9]+)/([0-9]+)/', FortConsumer),
-            re_path(r'ws/webssh/([0-9]+)/', SSHConsumer),
-            re_path(r'ws/fort_guacamole/([0-9]+)/([0-9]+)/(?P<group_name>.*)/', GuacamoleConsumer),
-            re_path(r'ws/admin_guacamole/([0-9]+)/(?P<group_name>.*)/', AdminGuacamole),
-        ]),
-    ),
-})
+# -*- coding: utf-8 -*-
+"""
+-------------------------------------------------
+   File Name：      routing
+   Description:
+   Author:          pythonzm
+   date：           2018/6/6
+-------------------------------------------------
+   Change Activity:
+                    2018/6/6:
+-------------------------------------------------
+"""
+from django.urls import path, re_path
+from channels.routing import ProtocolTypeRouter, URLRouter
+from channels.auth import AuthMiddlewareStack
+from projs.utils.log_websocket import LogConsumer
+from fort.utils.webssh import FortConsumer
+from assets.utils.webssh import SSHConsumer
+from assets.utils.docker_ssh import DockerConsumer
+from fort.utils.webguacamole import GuacamoleConsumer
+from assets.utils.webguacamole import AdminGuacamole
+from task.utils.ans_module_websocket import AnsModuleConsumer
+from task.utils.ans_playbook_websocket import AnsPlaybookConsumer
+from projs.utils.deploy_websocket import DeployConsumer
+
+application = ProtocolTypeRouter({
+
+    "websocket": AuthMiddlewareStack(
+        URLRouter([
+            # URLRouter just takes standard Django path() or url() entries.
+            path(r'ws/deploy/', DeployConsumer),
+            path(r'ws/ans_module_log/', AnsModuleConsumer),
+            path(r'ws/ans_playbook_log/', AnsPlaybookConsumer),
+            path(r'ws/deploy_log/', LogConsumer),
+            path(r'ws/dockerssh/', DockerConsumer),
+            re_path(r'ws/fortssh/([0-9]+)/([0-9]+)/', FortConsumer),
+            re_path(r'ws/webssh/([0-9]+)/', SSHConsumer),
+            re_path(r'ws/fort_guacamole/([0-9]+)/([0-9]+)/(?P<group_name>.*)/', GuacamoleConsumer),
+            re_path(r'ws/admin_guacamole/([0-9]+)/(?P<group_name>.*)/', AdminGuacamole),
+        ]),
+    ),
+})

Ops/settings.py

@@ -127,7 +127,7 @@
     'django.contrib.messages.middleware.MessageMiddleware',
     'django.middleware.clickjacking.XFrameOptionsMiddleware',
     'utils.middleware.UserLoginMiddleware',
-    'utils.middleware.RecordMiddleware',
+    # 'utils.middleware.RecordMiddleware',
 ]
 
 REST_FRAMEWORK = {

assets/urls.py

@@ -1,23 +1,24 @@
-# -*- coding: utf-8 -*-
-
-from django.urls import path, re_path
-from assets import views
-
-urlpatterns = [
-    path(r'assets_charts/', views.get_assets_charts, name='assets_charts'),
-    path(r'assets_list/', views.get_assets_list, name='assets_list'),
-    path(r'update_pwd/', views.update_pwd, name='update_pwd'),
-    path(r'add_asset/', views.add_asset, name='add_asset'),
-    path(r'add_base_asset/', views.add_base_asset, name='add_base_asset'),
-    re_path(r'update_asset/(?P<asset_type>(.*?))/(?P<pk>[0-9]+)/', views.update_asset, name='update_asset'),
-    path(r'server_facts/', views.server_facts, name='server_facts'),
-    re_path(r'get_server_info/(?P<pk>[0-9]+)/', views.get_asset_info, name='get_asset_info'),
-    path(r'import_assets/', views.import_assets, name='import_assets'),
-    re_path(r'export_assets/', views.export_assets, name='export_assets'),
-    re_path(r'ssh/(?P<pk>[0-9]+)/', views.ssh_terminal, name='ssh_terminal'),
-    re_path(r'guacamole/(?P<pk>[0-9]+)/', views.guacamole_terminal, name='guacamole_terminal'),
-    path(r'login_record/', views.login_record, name='login_record'),
-    re_path(r'admin_play/(?P<pk>[0-9]+)/', views.admin_play, name='admin_play'),
-    re_path(r'monitor/(?P<pk>[0-9]+)/', views.monitor, name='monitor'),
-    re_path(r'get_top_data/(?P<pk>[0-9]+)/', views.get_top_data, name='get_top_data'),
-]
+# -*- coding: utf-8 -*-
+
+from django.urls import path, re_path
+from assets import views
+
+urlpatterns = [
+    path(r'assets_charts/', views.get_assets_charts, name='assets_charts'),
+    path(r'assets_list/', views.get_assets_list, name='assets_list'),
+    path(r'update_pwd/', views.update_pwd, name='update_pwd'),
+    path(r'add_asset/', views.add_asset, name='add_asset'),
+    path(r'add_base_asset/', views.add_base_asset, name='add_base_asset'),
+    path(r'docker_ssh/', views.docker_ssh, name='docker_ssh'),
+    re_path(r'update_asset/(?P<asset_type>(.*?))/(?P<pk>[0-9]+)/', views.update_asset, name='update_asset'),
+    path(r'server_facts/', views.server_facts, name='server_facts'),
+    re_path(r'get_server_info/(?P<pk>[0-9]+)/', views.get_asset_info, name='get_asset_info'),
+    path(r'import_assets/', views.import_assets, name='import_assets'),
+    re_path(r'export_assets/', views.export_assets, name='export_assets'),
+    re_path(r'ssh/(?P<pk>[0-9]+)/', views.ssh_terminal, name='ssh_terminal'),
+    re_path(r'guacamole/(?P<pk>[0-9]+)/', views.guacamole_terminal, name='guacamole_terminal'),
+    path(r'login_record/', views.login_record, name='login_record'),
+    re_path(r'admin_play/(?P<pk>[0-9]+)/', views.admin_play, name='admin_play'),
+    re_path(r'monitor/(?P<pk>[0-9]+)/', views.monitor, name='monitor'),
+    re_path(r'get_top_data/(?P<pk>[0-9]+)/', views.get_top_data, name='get_top_data'),
+]

assets/utils/docker_ssh.py

@@ -0,0 +1,86 @@
+# -*- coding: utf-8 -*-
+"""
+-------------------------------------------------
+   File Name：      ssh
+   Description:
+   Author:          pythonzm
+   date：           2019-11-05
+-------------------------------------------------
+   Change Activity:
+                    2019-11-05:
+-------------------------------------------------
+"""
+# -*- coding: utf-8 -*-
+
+import time
+import docker
+import paramiko
+import threading
+from socket import timeout
+
+from docker.models.containers import Container
+
+from conf.logger import fort_logger
+from django.http.request import QueryDict
+from channels.generic.websocket import WebsocketConsumer
+
+
+class MyThread(threading.Thread):
+    def __init__(self, sock):
+        super(MyThread, self).__init__()
+        self.sock = sock
+        self._stop_event = threading.Event()
+
+    def stop(self):
+        self._stop_event.set()
+
+    def run(self):
+        while not self._stop_event.is_set():
+            try:
+                data = self.sock.tty._sock.recv(1024)
+                if data:
+                    str_data = data.decode('utf-8', 'ignore')
+                    self.sock.send(str_data)
+                else:
+                    return
+            except timeout:
+                break
+        self.sock.send('\n由于长时间没有操作，连接已断开!', close=True)
+        self.sock.close()
+
+
+class DockerConsumer(WebsocketConsumer):
+    def __init__(self, *args, **kwargs):
+        super(DockerConsumer, self).__init__(*args, **kwargs)
+        self.client = docker.from_env()
+        self.t1 = MyThread(self)
+        # self.query = QueryDict(query_string=self.scope.get('query_string'), encoding='utf-8')
+        # self.height = int(self.query.get('height'))
+        # self.width = int(self.query.get('width'))
+        self.tty = None
+
+    def connect(self):
+        if self.scope["user"].is_anonymous:
+            self.close(code=1007)
+        else:
+            self.accept()
+
+        try:
+            resp = self.client.api.exec_create('ae89214140b5', cmd='/bin/bash', stdout=True, stderr=True, stdin=True,
+                                               tty=True)
+
+            self.tty = self.client.api.exec_start(
+                resp['Id'], detach=False, tty=True, stream=False, socket=True
+            )
+        except Exception as e:
+            self.send('通过web连接容器失败！原因：{}'.format(e), close=True)
+
+        self.t1.setDaemon(True)
+        self.t1.start()
+
+    def receive(self, text_data=None, bytes_data=None):
+        self.tty._sock.send(text_data.encode('utf-8'))
+
+    def disconnect(self, close_code):
+        self.client.close()
+        self.t1.stop()