user.billing: fix imports for process_subscription move

- enums: add PremiumType
This commit is contained in:
Luna Mendes 2018-11-22 05:00:20 -03:00
parent 3d852af438
commit d62e140d8d
4 changed files with 14 additions and 12 deletions

View File

@ -1,14 +1,19 @@
import json import json
import datetime
from enum import Enum from enum import Enum
from quart import Blueprint, jsonify, request, current_app as app from quart import Blueprint, jsonify, request, current_app as app
from logbook import Logger
from litecord.auth import token_check from litecord.auth import token_check
from litecord.schemas import validate from litecord.schemas import validate
from litecord.snowflake import snowflake_datetime, get_snowflake from litecord.snowflake import snowflake_datetime, get_snowflake
from litecord.errors import BadRequest from litecord.errors import BadRequest
from litecord.types import timestamp_ from litecord.types import timestamp_, HOURS
from litecord.enums import UserFlags, PremiumType
from litecord.blueprints.users import mass_user_update
log = Logger(__name__)
bp = Blueprint('users_billing', __name__) bp = Blueprint('users_billing', __name__)
@ -43,12 +48,6 @@ class PaymentStatus:
FAILED = 2 FAILED = 2
class PremiumType:
TIER_1 = 1
TIER_2 = 2
NONE = None
PLAN_ID_TO_TYPE = { PLAN_ID_TO_TYPE = {
'premium_month_tier_1': PremiumType.TIER_1, 'premium_month_tier_1': PremiumType.TIER_1,
'premium_month_tier_2': PremiumType.TIER_2, 'premium_month_tier_2': PremiumType.TIER_2,

View File

@ -11,10 +11,7 @@ from litecord.blueprints.user.billing import (
) )
from litecord.snowflake import snowflake_datetime from litecord.snowflake import snowflake_datetime
from litecord.types import MINUTES, HOURS from litecord.types import MINUTES
from litecord.enums import UserFlags
from litecord.blueprints.users import mass_user_update
log = Logger(__name__) log = Logger(__name__)

View File

@ -12,7 +12,7 @@ from .auth import check_password
from litecord.auth import hash_data, check_username_usage from litecord.auth import hash_data, check_username_usage
from litecord.blueprints.guild.mod import remove_member from litecord.blueprints.guild.mod import remove_member
from litecord.blueprints.user.billing import PremiumType from litecord.enums import PremiumType
from litecord.images import parse_data_uri from litecord.images import parse_data_uri
from litecord.permissions import base_permissions from litecord.permissions import base_permissions

View File

@ -163,3 +163,9 @@ class MessageNotifications(EasyEnum):
ALL = 0 ALL = 0
MENTIONS = 1 MENTIONS = 1
NOTHING = 2 NOTHING = 2
class PremiumType:
TIER_1 = 1
TIER_2 = 2
NONE = None