import mariadb
import pandas as pd
from datetime import datetime

# MariaDB에 연결 (MariaDB를 사용)
def connect_db():
    try:
        conn = mariadb.connect(
            user="steemit",
            password="비밀번호",
            host="서버주소",
            database="steemit_postings"
        )
        return conn
    except mariadb.Error as e:
        print(f"Error connecting to MariaDB: {e}")
        return None

# 로그 함수
def log(message):
    print(f"{datetime.now()} - {message}")

# 특정 태그가 'kr-m2e'인 게시물 조회 함수
def fetch_kr_m2e_posts(conn):
    try:
        # 사용자별로 'kr-m2e' 메인 태그를 가진 게시물 조회 (최근 7개의 글만)
        query = """
            SELECT user_id, post_id, title, posting_date 
            FROM postings 
            WHERE main_tag = 'kr-m2e'
            ORDER BY user_id, posting_date DESC
        """
        df = pd.read_sql(query, conn)
        return df
    except Exception as e:
        log(f"An error occurred while fetching posts: {e}")
        return None

# Markdown 생성 함수
def generate_markdown(df):
    if df is None or df.empty:
        return "No posts found with the 'kr-m2e' tag."

    markdown_lines = ["# Posts with 'kr-m2e' Tag (Recent 7 posts per user)", ""]
    
    # 사용자별로 정리하여 최근 7개의 글만 표시
    grouped = df.groupby('user_id')
    for user_id, group in grouped:
        markdown_lines.append(f"## @{user_id}")
        recent_posts = group.head(7)  # 각 사용자별 최근 7개 글 선택
        for _, row in recent_posts.iterrows():
            post_url = row['post_id']
            title = row['title']
            posting_date = row['posting_date']
            markdown_lines.append(f"- [{title}]({post_url}) (Posted on: {posting_date})")
        markdown_lines.append("")  # 사용자별 빈 줄 추가
    
    return "\n".join(markdown_lines)

def save_markdown(content):
    # 현재 날짜를 파일 이름에 추가
    today_str = datetime.now().strftime('%Y-%m-%d')
    filename = f"kr-m2e-posting-{today_str}.md"
    
    try:
        with open(filename, 'w', encoding='utf-8') as f:
            f.write(content)
        log(f"Markdown file '{filename}' has been created successfully.")
    except Exception as e:
        log(f"An error occurred while writing the markdown file: {e}")

def main():
    # MariaDB 연결
    conn = connect_db()
    if conn is None:
        return

    # 'kr-m2e' 메인 태그가 포함된 게시물 조회
    df = fetch_kr_m2e_posts(conn)

    # Markdown 생성
    markdown_content = generate_markdown(df)

    # 결과를 파일로 저장
    save_markdown(markdown_content)

    # 연결 종료
    conn.close()

if __name__ == "__main__":
    main()

코드 실행하기

파일 저장하기

select_posting_m2e.py

위와 같이 파일을 저장 합니다.

파이썬 가상환경 실행하기

conda activate steemitm2e

프로그램 실행하기

python select_posting_m2e.py

맺음말

이번에도 DuckDB에서 MariaDB로 데이터베이스를 변경하는 작업을 했습니다.

그리고, 불러오는 데이터를 #kr-m2e 관련 데이터로 만들어서 진행 합니다.

다음 글에서는 수집하고, 정리한 글을 자동으로 포스팅 하는 프로그램을 생성하겠습니다.

감사합니다.

Posted through the ECblog app (https://blog.etain.club)

kr kr-dev kr-m2e talkit steemit stepn-kr ecblog

$1.00

431

Comments

Sort byBest

New to Nutbox?

[개발이야기#043] 내가 해보고 싶은 것 - 한국 스팀잇 M2E 사용자 모임 글 수집하기 - 게시글 정리하기

1 comment

관련글

머릿말

게시글 정리용 코드

코드 실행하기

파일 저장하기

파이썬 가상환경 실행하기

프로그램 실행하기

맺음말

Comments