📚 기능에 따라 코드정리, 웹크롤링

아직은 봇에 많은 기능이 생긴건 아니지만 나중을 위해서도 코드를 정리하는게 나을거같다고 생각해서 찾아보다가 Cog를 사용하여 정리할 수 있다해서 해봤는데

어떤 식으로 할지 여러 자료들을 찾다가 event를 사용하면 한정적인 코드밖에 못 짜기 때문에 어느정도 숙련된다면 event를 버리라는 조언의 글을 보았다.

지금까지 짠 코드에는 event가 투성이였기 때문에 기능에 따라 정리하는 김에 갈아엎기로 결정하였다.

#run.py

#run.py
import discord, asyncio, datetime, pytz, os
from discord.ext import commands
import os

client = commands.Bot(command_prefix="!") # 접두사를 설정해준다.

for filename in os.listdir("basics"):
    if filename.endswith(".py"):
        client.load_extension(f"basics.{filename[:-3]}")

@client.event
async def on_ready():
    print("고먐미 봇이 실행됩니다.")
    await client.change_presence(status=discord.Status.online, activity=discord.Game("고먐미 봇은 " + str(len(client.guilds)) + "개의 서버에 접속하고 있어요!")) # 봇의 상태메세지를 설정한다.

# 봇을 실행시키기 위해서 봇의 토큰을 작성하는 곳
client.run("토큰 값")

메인 코드는 이런식으로 짰다.

어차피 이미 짜놓은 코드가 있기 때문에 금방 짤 줄 알았는데, 파이썬을 안한지도 오래되고 자료들을 찾는다고 다소 시간이 걸렸다.

#simple.py

import discord, asyncio, datetime, pytz
from discord.ext import commands
from discord.utils import get
from discord import Member


client = commands.Bot(command_prefix='!')

class simple(commands.Cog): 

    def __init__(self, app):
        self.app = app 
    
    @commands.command(name="인사") # '!인사' 를 입력한다면
    async def hi(self, ctx): 
        await ctx.send("{} | {} 님 안녕하세요!".format(ctx.author, ctx.author.mention)) # 작성된 채널에 메세지를 출력한다.
        await ctx.author.send("{} | {} 님 안녕하세요!".format(ctx.author, ctx.author.mention)) # 작성한 유저에게 DM으로 메세지를 출력한다.

    @commands.command(name="정보") # '!정보'를 입력한다면
    async def information(self,ctx):
        embed = discord.Embed(title="고먐미", description="프로젝트용으로 개발중인 봇",timestamp=datetime.datetime.now(pytz.timezone('UTC')), color=0x4c2896)
        embed.add_field(name="다양한 기능들 추가 예정", value="시간날 떄마다 추가할 예정입니다.", inline=False)
        embed.add_field(name="GitHub", value="https://github.com/eastminn", inline=False)
        embed.set_footer(text="Bot Made by 김동민#0026, 문의는 DM으로 부탁드립니다 💬", icon_url="https://cdn.discordapp.com/attachments/955360993729449987/955392163158569021/IMG_20220320_042000_013.jpg")
        embed.set_thumbnail(url="https://cdn.discordapp.com/attachments/794866894615085087/956481187780644935/99pk9saw36t7q8t3o0gt.jpg")
        await ctx.send(embed=embed)

대충 이런식으로 지금은 간단한 기능만 있어서 한 파일에 코드를 다 적었다.

중간에 간단하게 해결될 문제인데 계속 잡고 있어서 대략 4시간 정도 걸린거같다.

다른 부분은 거의 똑같이 만들고, 인증 기능 부분만 어느정도 수정을 했다. 원래는 권한에 따라 오류메세지도 출력하게 하려했는데
계속 막혀서 다른 부분부터 만들고 나중에 손을 봐야할 거 같다.

새로운 기능을 만들고싶어서 최근에 많이 본 웹크롤링을 활용해서 기능을 만들기로 했다.

무난한 주제를 찾다가 코로나 19에 대한 정보를 크롤링해서 새로운 기능을 추가했다.

여기서 발생현황에 있는 일일 확진, 재원 위중증, 신규 입원, 일일 사망의 정보를 가져왔다. 가져오고 싶은 정보에 마우스를 대고 우클릭을 해서 검사를 누르고 검사창을 보고 코드를 짜면 된다.

@commands.command(name="코로나")
    async def crawl(self, ctx):
        url = "https://search.naver.com/search.naver?where=nexearch&sm=tab_etc&qvt=0&query=%EC%BD%94%EB%A1%9C%EB%82%9819"
        response = requests.get(url)  # get 방식으로 웹 정보 받아오기
        response_code = int(response.status_code)  # 응답 코드 받기
        
        if response_code == 200:  # 정상 작동(코드 200 반환) 시
            soup = BeautifulSoup(response.content, 'lxml')
            
        else:  # 오류 발생
            await ctx.send("웹 페이지 오류입니다.")
            
        # element
        new = soup.find("div", {"class":"status_info"}).findAll("p", {"class":"info_num"})
        today_infection = new[0].text # 일일 감염자
        critical_condition = new[1].text # 재원 위중증
        new_Hospitalization = new[2].text # 신규 입원
        today_died = new[3].text # 일일 사망자

        # 가독성을 위한 임베드
        embed = discord.Embed(title="🧬 코로나 19", description="http://ncov.mohw.go.kr/ 의 정보를 가져옵니다.",timestamp=datetime.datetime.now(pytz.timezone('UTC')), color=0x4c2896)
        embed.set_author(name="코로나 바이러스 감염증 - 19 (COVID-19)", url="http://ncov.mohw.go.kr/", icon_url="https://cdn.discordapp.com/attachments/779326874026377226/789094221533282304/Y5Tg6aur.png")
        embed.add_field(name="누적 확진자", value="{}명".format(cumulative_infection), inline=True)
        embed.add_field(name="일일 확진자", value="{}명".format(today_infection), inline=True)
        embed.add_field(name="재원 위중증", value="{}명".format(critical_condition), inline=True)
        embed.add_field(name="신규 입원", value="{}명".format(new_Hospitalization), inline=True)
        embed.add_field(name="일일 사망자", value="{}명".format(today_died), inline=True)
        embed.set_thumbnail(url="https://cdn.discordapp.com/attachments/956550003755741245/956610477184659486/9068ce115fb4499b9fc404664274e95e.png")
        embed.set_footer(text="자세한 정보는 위의 프로필 버튼의 사이트를 방문해주세요.", icon_url="https://cdn.discordapp.com/attachments/779326874026377226/789098688315654164/Flag_of_South_Korea.png")
        await ctx.send(embed=embed)

대충 코드는 BeautifulSoup 모듈을 사용해서 짰고 너무 어슬퍼서 나중에 추가할게 있으면 추가할 예정이다.
크롤링 코드는 웹페이지가 바뀔 때마다 코드가 제대로 작동하지 않을 수 있으므로 지속적으로 유지보수를 해줘야한다.