mirror of
https://github.com/ansible/awx.git
synced 2026-07-30 09:29:54 -02:30
fix flake8
This commit is contained in:
committed by
Ryan Petrello
parent
e25bd931a1
commit
093d204d19
@@ -41,6 +41,7 @@ class MetricsView(APIView):
|
|||||||
return Response(metrics().decode('UTF-8'))
|
return Response(metrics().decode('UTF-8'))
|
||||||
raise PermissionDenied()
|
raise PermissionDenied()
|
||||||
|
|
||||||
|
|
||||||
class BroadcastWebsocketMetricsView(APIView):
|
class BroadcastWebsocketMetricsView(APIView):
|
||||||
name = _('Broadcast Websockets')
|
name = _('Broadcast Websockets')
|
||||||
swagger_topic = 'Broadcast Websockets'
|
swagger_topic = 'Broadcast Websockets'
|
||||||
|
|||||||
@@ -1,8 +1,6 @@
|
|||||||
import datetime
|
import datetime
|
||||||
import os
|
|
||||||
import asyncio
|
import asyncio
|
||||||
import logging
|
import logging
|
||||||
import json
|
|
||||||
import aioredis
|
import aioredis
|
||||||
import redis
|
import redis
|
||||||
|
|
||||||
@@ -11,10 +9,7 @@ from prometheus_client import (
|
|||||||
Gauge,
|
Gauge,
|
||||||
Counter,
|
Counter,
|
||||||
Enum,
|
Enum,
|
||||||
Histogram,
|
|
||||||
Enum,
|
|
||||||
CollectorRegistry,
|
CollectorRegistry,
|
||||||
parser,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
@@ -118,7 +113,7 @@ class BroadcastWebsocketStats():
|
|||||||
'Number of messages received, to be forwarded, by the broadcast websocket system',
|
'Number of messages received, to be forwarded, by the broadcast websocket system',
|
||||||
registry=self._registry)
|
registry=self._registry)
|
||||||
self._messages_received = Gauge(f'awx_{self.remote_name}_messages_received',
|
self._messages_received = Gauge(f'awx_{self.remote_name}_messages_received',
|
||||||
'Number of messages received, to be forwarded, by the broadcast websocket system, for the duration of the current connection',
|
'Number forwarded messages received by the broadcast websocket system, for the duration of the current connection',
|
||||||
registry=self._registry)
|
registry=self._registry)
|
||||||
self._connection = Enum(f'awx_{self.remote_name}_connection',
|
self._connection = Enum(f'awx_{self.remote_name}_connection',
|
||||||
'Websocket broadcast connection',
|
'Websocket broadcast connection',
|
||||||
@@ -160,6 +155,5 @@ class BroadcastWebsocketStats():
|
|||||||
def serialize(self):
|
def serialize(self):
|
||||||
self.render()
|
self.render()
|
||||||
|
|
||||||
data = {}
|
|
||||||
registry_data = generate_latest(self._registry).decode('UTF-8')
|
registry_data = generate_latest(self._registry).decode('UTF-8')
|
||||||
return registry_data
|
return registry_data
|
||||||
|
|||||||
@@ -1,15 +1,9 @@
|
|||||||
|
|
||||||
import os
|
|
||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
import codecs
|
|
||||||
import datetime
|
import datetime
|
||||||
import hmac
|
import hmac
|
||||||
import asyncio
|
import asyncio
|
||||||
|
|
||||||
from django.utils.encoding import force_bytes
|
|
||||||
from django.utils.encoding import smart_str
|
|
||||||
from django.http.cookie import parse_cookie
|
|
||||||
from django.core.serializers.json import DjangoJSONEncoder
|
from django.core.serializers.json import DjangoJSONEncoder
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.utils.encoding import force_bytes
|
from django.utils.encoding import force_bytes
|
||||||
@@ -19,8 +13,6 @@ from channels.generic.websocket import AsyncJsonWebsocketConsumer
|
|||||||
from channels.layers import get_channel_layer
|
from channels.layers import get_channel_layer
|
||||||
from channels.db import database_sync_to_async
|
from channels.db import database_sync_to_async
|
||||||
|
|
||||||
from asgiref.sync import async_to_sync
|
|
||||||
|
|
||||||
|
|
||||||
logger = logging.getLogger('awx.main.consumers')
|
logger = logging.getLogger('awx.main.consumers')
|
||||||
XRF_KEY = '_auth_user_xrf'
|
XRF_KEY = '_auth_user_xrf'
|
||||||
@@ -53,8 +45,6 @@ class WebsocketSecretAuthHelper:
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def verify_secret(cls, s, nonce_tolerance=300):
|
def verify_secret(cls, s, nonce_tolerance=300):
|
||||||
hex_decoder = codecs.getdecoder("hex_codec")
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
(prefix, payload) = s.split(' ')
|
(prefix, payload) = s.split(' ')
|
||||||
if prefix != 'HMAC-SHA256':
|
if prefix != 'HMAC-SHA256':
|
||||||
@@ -160,9 +150,7 @@ class EventConsumer(AsyncJsonWebsocketConsumer):
|
|||||||
XRF_KEY not in self.scope["session"] or
|
XRF_KEY not in self.scope["session"] or
|
||||||
xrftoken != self.scope["session"][XRF_KEY]
|
xrftoken != self.scope["session"][XRF_KEY]
|
||||||
):
|
):
|
||||||
logger.error(
|
logger.error(f"access denied to channel, XRF mismatch for {user.username}")
|
||||||
"access denied to channel, XRF mismatch for {}".format(user.username)
|
|
||||||
)
|
|
||||||
await self.send_json({"error": "access denied to channel"})
|
await self.send_json({"error": "access denied to channel"})
|
||||||
return
|
return
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
import logging
|
import logging
|
||||||
import socket
|
|
||||||
import string
|
import string
|
||||||
import random
|
import random
|
||||||
import json
|
import json
|
||||||
|
|||||||
@@ -2,13 +2,11 @@ import inspect
|
|||||||
import logging
|
import logging
|
||||||
import sys
|
import sys
|
||||||
import json
|
import json
|
||||||
import re
|
|
||||||
from uuid import uuid4
|
from uuid import uuid4
|
||||||
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.db import connection
|
|
||||||
|
|
||||||
from . import pg_bus_conn, get_local_queuename
|
from . import pg_bus_conn
|
||||||
|
|
||||||
logger = logging.getLogger('awx.main.dispatch')
|
logger = logging.getLogger('awx.main.dispatch')
|
||||||
|
|
||||||
|
|||||||
@@ -7,7 +7,6 @@ import signal
|
|||||||
import sys
|
import sys
|
||||||
import redis
|
import redis
|
||||||
import json
|
import json
|
||||||
import re
|
|
||||||
import psycopg2
|
import psycopg2
|
||||||
from uuid import UUID
|
from uuid import UUID
|
||||||
from queue import Empty as QueueEmpty
|
from queue import Empty as QueueEmpty
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import logging
|
|||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.core.cache import cache as django_cache
|
from django.core.cache import cache as django_cache
|
||||||
from django.core.management.base import BaseCommand
|
from django.core.management.base import BaseCommand
|
||||||
from django.db import connection as django_connection, connections
|
from django.db import connection as django_connection
|
||||||
|
|
||||||
from awx.main.utils.handlers import AWXProxyHandler
|
from awx.main.utils.handlers import AWXProxyHandler
|
||||||
from awx.main.dispatch import get_local_queuename, reaper
|
from awx.main.dispatch import get_local_queuename, reaper
|
||||||
|
|||||||
@@ -4,7 +4,6 @@
|
|||||||
# Python
|
# Python
|
||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
import os
|
|
||||||
import redis
|
import redis
|
||||||
|
|
||||||
# Django
|
# Django
|
||||||
|
|||||||
@@ -1,4 +1,3 @@
|
|||||||
from django.urls import re_path
|
|
||||||
from django.conf.urls import url
|
from django.conf.urls import url
|
||||||
from channels.auth import AuthMiddlewareStack
|
from channels.auth import AuthMiddlewareStack
|
||||||
from channels.routing import ProtocolTypeRouter, URLRouter
|
from channels.routing import ProtocolTypeRouter, URLRouter
|
||||||
|
|||||||
@@ -1,18 +1,12 @@
|
|||||||
|
|
||||||
import os
|
|
||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
import asyncio
|
import asyncio
|
||||||
import datetime
|
|
||||||
import sys
|
|
||||||
|
|
||||||
import aiohttp
|
import aiohttp
|
||||||
from aiohttp import client_exceptions
|
from aiohttp import client_exceptions
|
||||||
|
|
||||||
from channels_redis.core import RedisChannelLayer
|
|
||||||
from channels.layers import get_channel_layer
|
from channels.layers import get_channel_layer
|
||||||
|
|
||||||
from django.utils.encoding import force_bytes
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.apps import apps
|
from django.apps import apps
|
||||||
from django.core.serializers.json import DjangoJSONEncoder
|
from django.core.serializers.json import DjangoJSONEncoder
|
||||||
|
|||||||
@@ -1,10 +1,8 @@
|
|||||||
import time
|
|
||||||
import threading
|
import threading
|
||||||
import logging
|
import logging
|
||||||
import atexit
|
import atexit
|
||||||
import json
|
import json
|
||||||
import ssl
|
import ssl
|
||||||
from datetime import datetime
|
|
||||||
|
|
||||||
from six.moves.queue import Queue, Empty
|
from six.moves.queue import Queue, Empty
|
||||||
from six.moves.urllib.parse import urlparse
|
from six.moves.urllib.parse import urlparse
|
||||||
@@ -187,7 +185,6 @@ class WSClient(object):
|
|||||||
self._send(json.dumps(payload))
|
self._send(json.dumps(payload))
|
||||||
|
|
||||||
def unsubscribe(self, wait=True, timeout=10):
|
def unsubscribe(self, wait=True, timeout=10):
|
||||||
time_start = datetime.now()
|
|
||||||
if wait:
|
if wait:
|
||||||
# Other unnsubscribe events could have caused the edge to trigger.
|
# Other unnsubscribe events could have caused the edge to trigger.
|
||||||
# This way the _next_ event will trigger our waiting.
|
# This way the _next_ event will trigger our waiting.
|
||||||
|
|||||||
Reference in New Issue
Block a user