mirror of https://gitlab.com/litecord/litecord.git
move create_guild_settings to common.guilds
This commit is contained in:
parent
c765cd7fe0
commit
7c3e9ec2c7
|
|
@ -21,7 +21,12 @@ from typing import Optional, List
|
||||||
|
|
||||||
from quart import Blueprint, request, current_app as app, jsonify
|
from quart import Blueprint, request, current_app as app, jsonify
|
||||||
|
|
||||||
from litecord.common.guilds import create_role, create_guild_channel, delete_guild
|
from litecord.common.guilds import (
|
||||||
|
create_role,
|
||||||
|
create_guild_channel,
|
||||||
|
delete_guild,
|
||||||
|
create_guild_settings,
|
||||||
|
)
|
||||||
|
|
||||||
from ..auth import token_check
|
from ..auth import token_check
|
||||||
from ..snowflake import get_snowflake
|
from ..snowflake import get_snowflake
|
||||||
|
|
@ -44,34 +49,6 @@ DEFAULT_EVERYONE_PERMS = 104324161
|
||||||
bp = Blueprint("guilds", __name__)
|
bp = Blueprint("guilds", __name__)
|
||||||
|
|
||||||
|
|
||||||
async def create_guild_settings(guild_id: int, user_id: int):
|
|
||||||
"""Create guild settings for the user
|
|
||||||
joining the guild."""
|
|
||||||
|
|
||||||
# new guild_settings are based off the currently
|
|
||||||
# set guild settings (for the guild)
|
|
||||||
m_notifs = await app.db.fetchval(
|
|
||||||
"""
|
|
||||||
SELECT default_message_notifications
|
|
||||||
FROM guilds
|
|
||||||
WHERE id = $1
|
|
||||||
""",
|
|
||||||
guild_id,
|
|
||||||
)
|
|
||||||
|
|
||||||
await app.db.execute(
|
|
||||||
"""
|
|
||||||
INSERT INTO guild_settings
|
|
||||||
(user_id, guild_id, message_notifications)
|
|
||||||
VALUES
|
|
||||||
($1, $2, $3)
|
|
||||||
""",
|
|
||||||
user_id,
|
|
||||||
guild_id,
|
|
||||||
m_notifs,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
async def add_member(guild_id: int, user_id: int):
|
async def add_member(guild_id: int, user_id: int):
|
||||||
"""Add a user to a guild."""
|
"""Add a user to a guild."""
|
||||||
await app.db.execute(
|
await app.db.execute(
|
||||||
|
|
|
||||||
|
|
@ -28,7 +28,6 @@ from ..auth import token_check
|
||||||
from ..schemas import validate, INVITE
|
from ..schemas import validate, INVITE
|
||||||
from ..enums import ChannelType
|
from ..enums import ChannelType
|
||||||
from ..errors import BadRequest, Forbidden
|
from ..errors import BadRequest, Forbidden
|
||||||
from .guilds import create_guild_settings
|
|
||||||
from ..utils import async_map
|
from ..utils import async_map
|
||||||
|
|
||||||
from litecord.blueprints.checks import (
|
from litecord.blueprints.checks import (
|
||||||
|
|
@ -39,6 +38,7 @@ from litecord.blueprints.checks import (
|
||||||
)
|
)
|
||||||
|
|
||||||
from litecord.blueprints.dm_channels import gdm_is_member, gdm_add_recipient
|
from litecord.blueprints.dm_channels import gdm_is_member, gdm_add_recipient
|
||||||
|
from litecord.common.guilds import create_guild_settings
|
||||||
|
|
||||||
log = Logger(__name__)
|
log = Logger(__name__)
|
||||||
bp = Blueprint("invites", __name__)
|
bp = Blueprint("invites", __name__)
|
||||||
|
|
|
||||||
|
|
@ -204,3 +204,31 @@ async def delete_guild(guild_id: int):
|
||||||
# becomes the little memer that tries to fuck up with
|
# becomes the little memer that tries to fuck up with
|
||||||
# everybody's gateway
|
# everybody's gateway
|
||||||
await app.dispatcher.remove("guild", guild_id)
|
await app.dispatcher.remove("guild", guild_id)
|
||||||
|
|
||||||
|
|
||||||
|
async def create_guild_settings(guild_id: int, user_id: int):
|
||||||
|
"""Create guild settings for the user
|
||||||
|
joining the guild."""
|
||||||
|
|
||||||
|
# new guild_settings are based off the currently
|
||||||
|
# set guild settings (for the guild)
|
||||||
|
m_notifs = await app.db.fetchval(
|
||||||
|
"""
|
||||||
|
SELECT default_message_notifications
|
||||||
|
FROM guilds
|
||||||
|
WHERE id = $1
|
||||||
|
""",
|
||||||
|
guild_id,
|
||||||
|
)
|
||||||
|
|
||||||
|
await app.db.execute(
|
||||||
|
"""
|
||||||
|
INSERT INTO guild_settings
|
||||||
|
(user_id, guild_id, message_notifications)
|
||||||
|
VALUES
|
||||||
|
($1, $2, $3)
|
||||||
|
""",
|
||||||
|
user_id,
|
||||||
|
guild_id,
|
||||||
|
m_notifs,
|
||||||
|
)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue