From e70d869b76f0e0c10db0a5d04ad22ead6658439e Mon Sep 17 00:00:00 2001 From: Casu Al Snek Date: Sat, 11 Jan 2025 23:14:35 +0545 Subject: [PATCH] Allow int casting of enums --- backend/blueprints/course/__init__.py | 8 ++++++-- backend/constants/__init__.py | 6 +++++- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/backend/blueprints/course/__init__.py b/backend/blueprints/course/__init__.py index 2f5a9ea..5c9782b 100644 --- a/backend/blueprints/course/__init__.py +++ b/backend/blueprints/course/__init__.py @@ -4,7 +4,7 @@ from werkzeug.datastructures import MultiDict import os import uuid from config import DEFAULT_COURSE_COVER -from ...db.model import db, Course, Category, User, Chat +from db.model import db, Course, Category, User, Chat from utils.utils import random_string_generator from utils.auth import auth_required, requires_role from constants import * @@ -120,4 +120,8 @@ def update_course(): db.session.commit() return jsonify({'message': 'Course info updated'}), 200 else: - return jsonify({'message': 'Unauthorized for this change'}), 401 \ No newline at end of file + return jsonify({'message': 'Unauthorized for this change'}), 401 + +@course.route('/info/') +def course_info(course_uuid): + pass \ No newline at end of file diff --git a/backend/constants/__init__.py b/backend/constants/__init__.py index 1ec55d3..3e2a32e 100644 --- a/backend/constants/__init__.py +++ b/backend/constants/__init__.py @@ -15,10 +15,14 @@ class PublishedStatus(Enum): REVOKED = 3 BANNED = 4 DRAFT = 5 + def __int__(self): + return self.value class NotificationTypes(Enum): MENTION = 0 COURSE_PUBLISH_STATUS_UPDATE = 1 NEW_BADGE = 2 TEXT_WITH_URL = 3 - PLAINTEXT_NOTICE = 4 \ No newline at end of file + PLAINTEXT_NOTICE = 4 + def __int__(self): + return self.value