python局域网小网站

2026-05-25 18:47:24
发布于：广东
16阅读
0回复
0点赞
本内容由AI生成，并非本人制作

from flask import Flask, request, session, redirect, url_for, render_template_string
import json
import os
import html
import threading
from datetime import datetime

app = Flask(__name__)
app.secret_key = os.urandom(24)

# ---------- 文件路径 ----------
CHAT_FILE = 'chat_data.json'
CODE_FILE = 'code_data.json'

# ---------- 线程锁 ----------
chat_lock = threading.Lock()
code_lock = threading.Lock()

# ---------- 文件读写 ----------
def load_json(filepath, default):
    try:
        with open(filepath, 'r', encoding='utf-8') as f:
            return json.load(f)
    except (FileNotFoundError, json.JSONDecodeError):
        return default

def save_json(filepath, data, lock):
    with lock:
        with open(filepath, 'w', encoding='utf-8') as f:
            json.dump(data, f, ensure_ascii=False, indent=2)

# ---------- 启动时加载 ----------
chat_messages = load_json(CHAT_FILE, [])
code_snippets = load_json(CODE_FILE, {})
next_code_id = max([int(k) for k in code_snippets.keys()], default=0) + 1 if isinstance(code_snippets, dict) else 1

# ---------- 主页 ----------
@app.route('/')
def index():
    html = """
    <!DOCTYPE html>
    <html>
    <head>
        <meta charset="utf-8">
        <title>局域网工具</title>
        <style>
            body { font-family: sans-serif; text-align: center; margin-top: 50px;
                   background: #f0f2f5; }
            a { display: inline-block; margin: 20px; padding: 15px 35px;
                font-size: 18px; background: #4CAF50; color: white;
                text-decoration: none; border-radius: 8px;
                box-shadow: 0 2px 6px rgba(0,0,0,0.15); }
            a:hover { background: #45a049; transform: scale(1.03); }
            h1 { color: #333; }
        </style>
    </head>
    <body>
        <h1>欢迎使用局域网工具</h1>
        <p style="color: #888;">数据已持久化到本地文件，重启不丢失</p>
        <a href="/liaotian">去聊天</a>
        <a href="/daima">传输代码</a>
    </body>
    </html>
    """
    return render_template_string(html)

# ---------- 聊天界面 ----------
@app.route('/liaotian', methods=['GET', 'POST'])
def liaotian():
    if 'name' not in session:
        session['name'] = '匿名'

    if request.method == 'POST':
        if 'new_name' in request.form:
            session['name'] = request.form['new_name'].strip() or '匿名'
        elif 'message' in request.form:
            msg = request.form['message'].strip()
            if msg:
                now = datetime.now().strftime('%m-%d %H:%M')
                chat_messages.append({
                    'name': session.get('name', '匿名'),
                    'message': msg,
                    'time': now
                })
                save_json(CHAT_FILE, chat_messages, chat_lock)
        return redirect(url_for('liaotian'))

    chat_html = ""
    for item in chat_messages:
        name_esc = html.escape(item['name'])
        msg_esc = html.escape(item['message'])
        time_str = html.escape(item.get('time', ''))
        chat_html += (
            f'<p><b>{name_esc}</b> '
            f'<span style="font-size: 12px; color: #999;">{time_str}</span>'
            f'<br>{msg_esc}</p>'
        )

    name_esc = html.escape(session.get('name', '匿名'))

    html_code = f"""
    <!DOCTYPE html>
    <html>
    <head>
        <meta charset="utf-8">
        <title>聊天室</title>
        <style>
            body {{ font-family: sans-serif; max-width: 600px; margin: 20px auto;
                   background: #f0f2f5; }}
            .chat-box {{ border: 1px solid #***; padding: 12px; height: 350px;
                         overflow-y: auto; margin-bottom: 10px; background: white;
                         border-radius: 6px; }}
            .chat-box p {{ margin: 8px 0; word-break: break-all; line-height: 1.5; }}
            input, button {{ padding: 8px 12px; border-radius: 4px; border: 1px solid #ccc; }}
            button {{ background: #4CAF50; color: white; border: none; cursor: pointer; }}
            button:hover {{ background: #45a049; }}
            .name-area {{ margin-bottom: 20px; background: white; padding: 12px;
                          border-radius: 6px; border: 1px solid #***; }}
            a {{ color: #4CAF50; }}
        </style>
    </head>
    <body>
        <h2>聊天室</h2>
        <div class="name-area">
            <form method="POST">
                你的名字:
                <input type="text" name="new_name" value="{name_esc}" style="width: 150px;">
                <button type="submit">更改</button>
            </form>
        </div>
        <div class="chat-box">{chat_html}</div>
        <form method="POST">
            <input type="text" name="message" placeholder="输入消息..." style="width: 70%;">
            <button type="submit">发送</button>
        </form>
        <br><a href="/">← 返回主页</a>
    </body>
    </html>
    """
    return render_template_string(html_code)

# ---------- 代码传输界面 ----------
@app.route('/daima', methods=['GET', 'POST'])
def daima():
    global next_code_id, code_snippets

    if request.method == 'POST':
        action = request.form.get('action')

        if action == 'view':
            code_id = request.form.get('code_id', '').strip()
            if code_id.isdigit():
                cid = int(code_id)
                snippet = code_snippets.get(str(cid))
                if snippet:
                    return render_template_string(code_display_html(cid, snippet))
                else:
                    return render_template_string(message_html("错误", "编号不存在，请检查后重试"))
            else:
                return render_template_string(message_html("错误", "请输入有效数字编号"))

        elif action == 'send':
            author = request.form.get('author', '').strip()
            code = request.form.get('code', '').strip()
            language = request.form.get('language', 'txt').strip()
            if not author or not code:
                return render_template_string(message_html("错误", "发布者和代码内容不能为空"))
            cid = next_code_id
            code_snippets[str(cid)] = {
                'author': author,
                'code': code,
                'language': language
            }
            save_json(CODE_FILE, code_snippets, code_lock)
            next_code_id += 1
            return render_template_string(
                message_html("发送成功", f"发送完毕，你的编号为 {cid}。其他人可用此编号查看代码。", show_back=True)
            )

    html = """
    <!DOCTYPE html>
    <html>
    <head>
        <meta charset="utf-8">
        <title>代码传输</title>
        <style>
            body { font-family: sans-serif; max-width: 500px; margin: 20px auto;
                   background: #f0f2f5; }
            .form-box { border: 1px solid #***; padding: 18px; margin: 12px 0;
                        background: white; border-radius: 6px; }
            input, textarea, button, select { padding: 8px 12px; border-radius: 4px;
                                               border: 1px solid #ccc; margin: 5px 0; }
            textarea { width: 96%; height: 120px; font-family: monospace; }
            input[type="text"] { width: 96%; }
            select { width: 100%; }
            button { background: #4CAF50; color: white; border: none; cursor: pointer;
                     padding: 10px; font-size: 15px; }
            button:hover { background: #45a049; }
            h3 { margin-top: 0; }
            a { color: #4CAF50; }
        </style>
    </head>
    <body>
        <h2>代码传输</h2>
        <div class="form-box">
            <h3>查看代码</h3>
            <form method="POST">
                <input type="hidden" name="action" value="view">
                编号: <input type="text" name="code_id" placeholder="输入编号，如 1">
                <button type="submit">查看代码</button>
            </form>
        </div>
        <div class="form-box">
            <h3>发送代码</h3>
            <form method="POST">
                <input type="hidden" name="action" value="send">
                发布者: <input type="text" name="author" placeholder="你的名字"><br>
                请选择语言:
                <select name="language">
                    <option value="txt">txt</option>
                    <option value="c++">c++</option>
                    <option value="python">python</option>
                    <option value="其他">其他</option>
                </select><br>
                代码:<br>
                <textarea name="code" placeholder="在此粘贴代码..."></textarea><br>
                <button type="submit" style="width: 100%;">发送代码</button>
            </form>
        </div>
        <br><a href="/">← 返回主页</a>
    </html>
    """
    return render_template_string(html)

# ---------- 辅助页面函数 ----------
def code_display_html(cid, snippet):
    author_esc = html.escape(snippet['author'])
    code_esc = html.escape(snippet['code'])
    lang = html.escape(snippet.get('language', 'txt'))

    return f"""
    <!DOCTYPE html>
    <html>
    <head><meta charset="utf-8"><title>代码 #{cid}</title>
    <style>
        body {{ font-family: sans-serif; max-width: 650px; margin: 20px auto; background: #f0f2f5; }}
        pre {{ background: #1e1e1e; color: #d4d4d4; padding: 16px; border-radius: 6px;
               white-space: pre-wrap; overflow-x: auto; font-size: 14px; }}
        .info {{ background: white; padding: 12px 16px; border-radius: 6px; margin-bottom: 12px; }}
        .info p {{ margin: 4px 0; }}
        .lang-tag {{ display: inline-block; background: #4CAF50; color: white;
                     padding: 2px 10px; border-radius: 12px; font-size: 13px; }}
        a {{ color: #4CAF50; }}
    </style>
    </head>
    <body>
        <div class="info">
            <h2>代码 #{cid}</h2>
            <p><b>发布者:</b> {author_esc}</p>
            <p><b>语言:</b> <span class="lang-tag">{lang}</span></p>
        </div>
        <pre>{code_esc}</pre>
        <br><a href="/daima">← 返回代码传输</a>
    </body>
    </html>
    """

def message_html(title, message, show_back=False):
    back = '<br><a href="/daima">← 返回代码传输</a>' if show_back else ''
    return f"""
    <!DOCTYPE html>
    <html>
    <head><meta charset="utf-8"><title>{title}</title>
    <style>
        body {{ font-family: sans-serif; max-width: 500px; margin: 50px auto;
               text-align: center; background: #f0f2f5; }}
        .msg-box {{ background: white; padding: 30px; border-radius: 8px;
                    border: 1px solid #***; }}
        a {{ color: #4CAF50; }}
    </style>
    </head>
    <body>
        <div class="msg-box">
            <h2>{title}</h2>
            <p style="font-size: 16px;">{message}</p>
            {back}
            <br><a href="/">← 返回主页</a>
        </div>
    </body>
    </html>
    """

# ---------- 启动 ----------
if __name__ == '__main__':
    print("正在启动服务器...")
    print(f"聊天记录 → {os.path.abspath(CHAT_FILE)}")
    print(f"代码库   → {os.path.abspath(CODE_FILE)}")
    app.run(host='0.0.0.0', port=5000, debug=True)