main.py

from nextcord import *
import db
import cmp
from cmp import emojis
cli = Client(activity=Game("TETR.IO"))

import datetime, time, math

@cli.event
async def on_ready():
    print("시작됨")


@cli.slash_command(name="가입", description="가입을 함")
async def sign_up(inter: Interaction):
    await inter.response.send_modal(cmp.modals.SignUp())

@cli.slash_command(name="새로고침", description="자신의 테트리오 정보를 다시 불러옴")
async def refresh(inter: Interaction):
    await inter.response.send_message("> 새로고침중")
    try: before = db.User().get(id=inter.user.id)
    except: return await inter.edit_original_message(content = "> 가입 먼저해야함")
    
    db.User().push(inter.user.id, before.nick, before.name)
    await inter.edit_original_message(content = "> 새로고침 완료")

@cli.slash_command(name="정보", description="테트리오 정보를 불러옴")
async def info(inter: Interaction):
    try: data = db.User().get(inter.user.id)
    except: return await inter.response.send_message(embed = Embed(title = "Not Founded", description="가입을 먼저 해야함", color = 0xff7033), ephemeral = True)
    
    await inter.response.defer()

    embeds = [Embed(title = f"{data.nick}(`{data.name}`)", color=0xa1dba5)]
    embeds[0].set_thumbnail(url=data.avatar_img)
    if data.name == data.nick: embeds[0].title = data.name
    embeds[0].add_field(name="이김/플레이", value=f"{data.gamewon}/{data.gameplayed}", inline = False)
    
    if data.l40.ok:
        embeds.append(Embed(title = f"40 Line", color=0xa1dba5))
        embeds[len(embeds)-1].add_field(name="시간", value=f"{int(data.l40.time/60)}분 {int(data.l40.time%60)}초", inline = False)
        ts = int(time.mktime(datetime.datetime.strptime(data.l40.ts, "%Y-%m-%dT%H:%M:%S").timetuple()))
        embeds[len(embeds)-1].add_field(name="~전에 완료", value=f"<t:{ts}:R>", inline = False)
    
    if data.blitz.ok:
        embeds.append(Embed(title = f"Blitz", color=0xa1dba5))
        embeds[len(embeds)-1].add_field(name="점수", value=f"{data.blitz.point:,}점")
        ts = int(time.mktime(datetime.datetime.strptime(data.l40.ts, "%Y-%m-%dT%H:%M:%S").timetuple()))
        embeds[len(embeds)-1].add_field(name="~전에 완료", value=f"<t:{ts}:R>", inline = False)
    

    if len(data.badges):
        description = ""
        for badge in data.badges:
            description += f"{emojis.badges.get(badge.title, badge.title)} "
        embeds.append(Embed(title = f"Badge", color=0xa1dba5, description = description))
    

    if data.league.rank != 'z':
        embeds.append(Embed(title = f"Tetra League", color=0xa1dba5))
        embeds[len(embeds)-1].add_field(name="레이팅", value=f"**{data.league.rating:,}**TR", inline = False)
        embeds[len(embeds)-1].add_field(name="랭크", value=f"{emojis.ranks[data.league.rank]}")
        embeds[len(embeds)-1].add_field(name="최고랭크", value=f"{emojis.ranks[data.league.best_rank]}")
        embeds[len(embeds)-1].add_field(name="apm (Attack Per Minute)", value=f"{data.league.apm}", inline = False)
        embeds[len(embeds)-1].add_field(name="pps (Pieces Per Seconds)", value=f"{data.league.pps}")
        embeds[len(embeds)-1].add_field(name="vs (VerSus)", value=f"{data.league.vs}%")
    
    
    
    embeds[len(embeds)-1].set_image(url=data.banner_img)

    await inter.followup.send(embeds = embeds)


@cli.slash_command(name="클랜생성", description="클랜을 생성함")
async def sign_up(inter: Interaction):
    try: data = db.User().get(inter.user.id)
    except: return await inter.response.send_message(embed = Embed(title = "Not Founded", description="가입을 먼저 해야함", color = 0xff7033), ephemeral = True)
    if data.clan != None:
        return await inter.response.send_message("이미 클랜에 있음")
    await inter.response.send_modal(cmp.modals.Clan("", "", "", ""))

@cli.slash_command(name="클랜참가", description="클랜에 참가함")
async def join_clan(inter: Interaction, name: str = SlashOption(name="클랜명", description="참가 할 클랜명")):
    try: data = db.User().get(inter.user.id).clan
    except: return await inter.response.send_message(embed = Embed(title = "Not Founded", description="가입을 먼저 해야함", color = 0xff7033), ephemeral = True)
    
    if data != None:
         return await inter.response.send_message(embed = Embed(title = "Clan Error", description="이미 참가한 클랜이 있음", color = 0xff7033), ephemeral = True)
    
    clan = db.Clan().get(name)

    if clan.pw == None:
        db.User().add_clan(inter.user.id, clan)
        await inter.response.send_message(embed = Embed(title = f"성공{emojis.verified}", description=f"[{clan.name}]에 가입을 성공함", color=0xa1dba5))
    else:
        await inter.response.send_modal(cmp.modals.SignUpClan(clan))        

@join_clan.on_autocomplete("name")
async def join_clan_auto(inter: Interaction, name: str):
    auto = ["입력"]
    if name != "":
        auto = [i for i in db.Clan().file if name.lower() in i]
        if auto == []:
            auto = ["검색결과 없음"]

    await inter.response.send_autocomplete(auto)

@cli.slash_command(name="클랜랭크", description="클랜에서 랭크를 보여줌")
async def clanrank(inter: Interaction, option: str = SlashOption(name="옵션", description="정렬할 옵션" , choices=["blitz", "40l", "tetra league"])):
    try: data = db.User().get(inter.user.id).clan
    except: return await inter.response.send_message(embed = Embed(title = "Not Founded", description="가입을 먼저 해야함", color = 0xff7033), ephemeral = True)
    
    if data == None:
        return await inter.response.send_message(embed = Embed(title = "Clan Error", description="클랜에 가입해야함", color = 0xff7033))
    
    await inter.response.defer()

    members = db.Clan().get(data).members
    
    arr = []
    embed = Embed(title=f"[{data}]클랜 {option} 랭킹", color=0xa1dba5)
    for member in members:
        user = db.User().get(member)
        if option == "40l":
            value = (user.l40.time, f"{int(user.l40.time/60)}분 {int(user.l40.time%60)}초", user)
            if not user.l40.ok:
                value = (99999, f"NaN", user)

        
        elif option == "blitz":
            value = (-user.blitz.point, f"{user.blitz.point:,}점", user)
            if not user.blitz.ok:
                value = (1, f"NaN점", user)
        else:
            value = (-user.league.rating, f"{emojis.ranks[user.league.rank]} **{user.league.rating:,}**TL", user)
            if user.league.rank == "z":
                value = (0, f"{emojis.ranks[user.league.rank]} **0**TL", user)
                
            ok = True

        # if ok:
        arr.append(value)
    
    arr.sort(key=lambda x: x[0])
    
    i = 1

    for rank in arr[0:5]:
        embed.add_field(name=f"{i}등 | {rank[2].nick}({rank[2].name})", value=rank[1], inline=False)
        i += 1

    await inter.followup.send(embed = embed, view=cmp.buttons.RankBtn(arr))


@cli.slash_command(name="클랜수정", description="클랜정보 수정")
async def change_clan(inter: Interaction):
    try: data = db.User().get(inter.user.id)
    except: return await inter.response.send_message(embed = Embed(title = "Not Founded", description="가입을 먼저 해야함", color = 0xff7033), ephemeral = True)
    
    if data.clan == None:
        return await inter.response.send_message(embed = Embed(title = "Not Founded", description="클랜이 없음", color = 0xff7033), ephemeral = True)
    clan = db.Clan().get(data.clan)
    
    if clan.admin != inter.user.id:
        return await inter.response.send_message(embed = Embed(title="Not Permission", description="너 어드민 아님", color=0xff7033), ephemeral=True)
    
    await inter.response.send_modal(cmp.modals.ClanChange(clan))


@cli.slash_command(name="클랜탈퇴", description="클랜에서 나감")
async def leave_clan(inter: Interaction):
    try: data = db.User().get(inter.user.id)
    except: return await inter.response.send_message(embed = Embed(title = "Not Founded", description="가입을 먼저 해야함", color = 0xff7033), ephemeral = True)
    
    if data.clan == None:
        return await inter.response.send_message(embed = Embed(title = "Not Founded", description="클랜이 없음", color = 0xff7033), ephemeral = True)
    clan = db.Clan().get(data.clan)
    
    if clan.admin == inter.user.id:
        return await inter.response.send_message(embed = Embed(title="Not Permission", description="너 어드민임", color=0xff7033), ephemeral=True)
    db.Clan().delete(db.User, clan.name)
    await inter.response.send_message(embed = Embed(title = "탈퇴 성공!", description = f"[{clan.name}]클랜을 성공적으로 변경", color=0xa1dba5))



@cli.slash_command(name="클랜정보", description="클랜 정보를 보여줌")
async def clan_info(inter: Interaction, name: str = SlashOption(name="클랜명", description="참가 할 클랜명", required=False, default=None)):
    if name == None:
        try: name = db.User().get(inter.user.id).clan
        except: return await inter.response.send_message(embed = Embed(title = "Not Founded", description="가입을 먼저 해야함", color = 0xff7033), ephemeral = True)
    
    if name == None:
         return await inter.response.send_message(embed = Embed(title = "Clan Error", description="참가한 클랜이 없음", color = 0xff7033), ephemeral = True)
    
    clan = db.Clan().get(name)
    user = db.User()
    embeds = [
        Embed(title = f"[{clan.name}]클랜", color=0x3366ff),
        Embed(title = f"레이팅", color=0x3366ff),
        Embed(title = f"솔로", color=0x3366ff),
        Embed(title = f"평균 속도", color=0x3366ff)
    ]
    rating = [0, 0]
    l40 = [0, 0]
    blitz = [0, 0]
    league = 0
    spd = [0, 0, 0]
    for member in clan.members:
        if user.get(member).league.rating != None:
            rating[0] += user.get(member).league.rating
            rating[1] += 1
        
        if user.get(member).l40.ok:
            l40[0] += user.get(member).l40.time
            l40[1] += 1

        if user.get(member).blitz.ok:
            blitz[0] += user.get(member).blitz.point
            blitz[1] += 1

        if user.get(member).league.rank != "z":
            league += 1
            spd[0] += user.get(member).league.apm
            spd[1] += user.get(member).league.pps
            spd[2] += user.get(member).league.vs

    embeds[0].add_field(name="설명", value=clan.des, inline=False)
    embeds[0].add_field(name="멤버수", value=len(clan.members), inline=False)
    
    embeds[1].add_field(name="합", value=f"**{rating[0]:,}**TL", inline=False)
    embeds[1].add_field(name="평균", value=f"**{rating[0]/rating[1]:,.2f}**TL", inline=False)
    embeds[1].add_field(name="제한", value=f"**{clan.rating:,}**TL", inline=False)
    
    embeds[2].add_field(name="blitz합", value=f"**{blitz[0]:,}**B", inline=False)
    embeds[2].add_field(name="blitz 평균", value=f"**{blitz[0]/blitz[1]:,.2f}**B", inline=False)
    embeds[2].add_field(name="40l합", value=f"{int(l40[0]/60)}분 {int(l40[0]%60)}초", inline=False)
    embeds[2].add_field(name="40l 평균", value=f"{int(l40[0]/l40[1]/60)}분 {int(l40[0]/l40[1]%60)}초", inline=False)
    
    embeds[3].add_field(name="apm(VerSus)", value=f"{spd[0]/league:.2f}", inline=False)
    embeds[3].add_field(name="pps(Pieces Per Seconds)", value=f"{spd[1]/league:.2f}", inline=False)
    embeds[3].add_field(name="vs(VerSus)", value=f"{spd[2]/league:.2f}%", inline=False)
    
    await inter.response.send_message(embeds = embeds)

@clan_info.on_autocomplete("name")
async def clan_info_auto(inter: Interaction, name: str):
    auto = ["입력"]
    if name != "":
        auto = [i for i in db.Clan().file if name.lower() in i]
        if auto == []:
            auto = ["검색결과 없음"]

    await inter.response.send_autocomplete(auto)




cli.run(open(".token").read())