From d41641929c1ec34a4cf6025c8b72b40e0cb0ce8c Mon Sep 17 00:00:00 2001 From: slice Date: Tue, 19 Jun 2018 17:52:25 -0700 Subject: [PATCH] add ws stuff to example config --- config.example.py | 2 ++ litecord/blueprints/auth.py | 4 ++-- litecord/gateway/errors.py | 2 +- litecord/gateway/websocket.py | 6 +++--- 4 files changed, 8 insertions(+), 6 deletions(-) diff --git a/config.example.py b/config.example.py index 19864d9..de6c35b 100644 --- a/config.example.py +++ b/config.example.py @@ -6,6 +6,8 @@ class Config: DEBUG = False IS_SSL = False WEBSERVER_URL = 'localhost:5000' + WS_HOST = 'localhost' + WS_PORT = 5001 POSTGRES = {} diff --git a/litecord/blueprints/auth.py b/litecord/blueprints/auth.py index 478d321..4624728 100644 --- a/litecord/blueprints/auth.py +++ b/litecord/blueprints/auth.py @@ -6,8 +6,8 @@ import itsdangerous import bcrypt from quart import Blueprint, jsonify, request, current_app as app -from ..snowflake import get_snowflake -from ..errors import AuthError +from litecord.snowflake import get_snowflake +from litecord.errors import AuthError bp = Blueprint('auth', __name__) diff --git a/litecord/gateway/errors.py b/litecord/gateway/errors.py index c74fade..e217d27 100644 --- a/litecord/gateway/errors.py +++ b/litecord/gateway/errors.py @@ -1,4 +1,4 @@ -from ..errors import WebsocketClose +from litecord.errors import WebsocketClose class UnknownOPCode(WebsocketClose): diff --git a/litecord/gateway/websocket.py b/litecord/gateway/websocket.py index 56e773f..3ee0637 100644 --- a/litecord/gateway/websocket.py +++ b/litecord/gateway/websocket.py @@ -4,13 +4,13 @@ import collections import earl -from ..errors import WebsocketClose, AuthError -from ..auth import raw_token_check +from litecord.errors import WebsocketClose, AuthError +from litecord.auth import raw_token_check from .errors import DecodeError, UnknownOPCode, \ InvalidShard, ShardingRequired from .opcodes import OP -from .state import GatewayState, gen_session_id +from .state import GatewayState log = logging.getLogger(__name__)