import sqlite3 import csv from io import StringIO from datetime import datetime, timedelta from flask import Flask, request, render_template_string, redirect, Response, session app = Flask(__name__) app.secret_key = 'super_secret_key_netmarble' DB_PATH = '/home/ajw103/mysite/visitors.db' def get_kst_now(): return datetime.utcnow() + timedelta(hours=9) # 🛠️ 1. 데이터베이스 준비 (방문목적, 출차시간 추가!) def init_db(): conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute('''CREATE TABLE IF NOT EXISTS visitors (name TEXT, visitor_company TEXT, car_number TEXT, host_info TEXT, purpose TEXT, visit_date TEXT, visit_time TEXT, status TEXT DEFAULT '대기중', parking_time TEXT, entry_time TEXT, exit_time TEXT)''') # 예전 금고에 없는 칸 자동 추가 마법 new_columns = ['visitor_company', 'host_info', 'purpose', 'visit_date', 'visit_time', 'parking_time', 'entry_time', 'exit_time'] for col in new_columns: try: c.execute(f"ALTER TABLE visitors ADD COLUMN {col} TEXT") except: pass conn.commit() conn.close() init_db() # 🔐 관리자 로그인 화면 login_html = """

🔒 관리자 로그인

{% if error %}

{% endif %}

""" # 👨‍💼 담당자 화면 staff_html = """

👨‍💼 담당자: 방문객 초대

➡️ 인포데스크 모니터 가기

""" # 📱 방문객 폼 화면 (방문 목적 추가!) visitor_html = """

🏢 {% if v %}방문 일정 변경{% else %}사전 방문신청{% endif %}

👤 방문자 성함 🏢 방문자 소속 🚘 차량 번호 (도보시 '없음') 🤝 방문 회사 및 담당자 🎯 방문 목적 📅 방문 예정 일자 ⏰ 방문 예정 시간

[개인정보 수집 및 이용 동의]
- 수집 항목: 성명, 소속, 차량번호
- 수집 목적: 방문객 식별 및 주차 등록
- 보유 기간: 방문 종료 후 1개월 뒤 파기

개인정보 수집 및 이용에 동의합니다. (필수)

""" # ✅ 신청 완료 화면 success_html = """

✅ 방문 신청이 완료되었습니다

입력하신 정보는 안전하게 저장되었습니다.
방문 당일 조심히 오세요!

✏️ 방문 일정 변경 / 확인하기

📍 오시는 길

🏢 서울 구로구 디지털로26길 38 지타워

""" # 🏢 4. 인포데스크 화면 (출차 시간, 방문 목적 열 추가) desk_html = """

🏢 관리자 대시보드

로그아웃

{% if visitors %} {% for v in visitors %} {% endfor %} {% else %} {% endif %}

예정 시간	입차 시간	출차 시간	방문자 성함 (소속)	차량 번호	방문 대상	방문 목적	상태	주차시간 등록	관리
⏰ {{ v['visit_time'] }}	{% if not v['entry_time'] %} 입차 전 {% else %} {{ v['entry_time'] }} {% endif %}	{% if not v['exit_time'] %} 출차 전 {% if v['entry_time'] %} {% endif %} {% else %} {{ v['exit_time'] }} {% endif %}	{{ v['name'] }} ({{ v['visitor_company'] }})	{{ v['car_number'] }}	{{ v['host_info'] }}	{{ v['purpose'] }}	{% if not v['parking_time'] %}미등록 {% else %}누적: {{ v['parking_time'] }}{% endif %}
해당 조건에 일치하는 방문객이 없습니다.

📥 엑셀(CSV) 다운로드

⬅️ 담당자 화면으로 돌아가기

""" # --- 라우팅 (서버 기능) --- @app.route('/login', methods=['GET', 'POST']) def login(): if request.method == 'POST': if request.form.get('admin_id') == 'netmarble' and request.form.get('admin_pw') == 'netmarble1!': session['logged_in'] = True return redirect('/desk') return render_template_string(login_html, error="아이디 또는 비밀번호가 틀렸습니다.") return render_template_string(login_html) @app.route('/logout') def logout(): session.pop('logged_in', None) return redirect('/login') @app.route('/') def staff_home(): return staff_html @app.route('/register') def register_form(): return render_template_string(visitor_html, v=None) @app.route('/submit', methods=['POST']) def submit(): conn = sqlite3.connect(DB_PATH) c = conn.cursor() # 방문 목적(purpose)을 저장하는 쿼리로 업데이트 c.execute("""INSERT INTO visitors (name, visitor_company, car_number, host_info, purpose, visit_date, visit_time) VALUES (?, ?, ?, ?, ?, ?, ?)""", (request.form.get("name"), request.form.get("visitor_company"), request.form.get("car_number"), request.form.get("host_info"), request.form.get("purpose"), request.form.get("visit_date"), request.form.get("visit_time"))) visitor_id = c.lastrowid conn.commit() conn.close() return render_template_string(success_html, visitor_id=visitor_id) @app.route('/edit_visitor/') def edit_visitor(vid): conn = sqlite3.connect(DB_PATH) conn.row_factory = sqlite3.Row c = conn.cursor() c.execute("SELECT rowid, * FROM visitors WHERE rowid = ?", (vid,)) v = c.fetchone() conn.close() return render_template_string(visitor_html, v=v) @app.route('/update_visitor/', methods=['POST']) def update_visitor(vid): conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute("""UPDATE visitors SET name=?, visitor_company=?, car_number=?, host_info=?, purpose=?, visit_date=?, visit_time=? WHERE rowid=?""", (request.form.get("name"), request.form.get("visitor_company"), request.form.get("car_number"), request.form.get("host_info"), request.form.get("purpose"), request.form.get("visit_date"), request.form.get("visit_time"), vid)) conn.commit() conn.close() return render_template_string(success_html, visitor_id=vid) @app.route('/desk') def desk(): if not session.get('logged_in'): return redirect('/login') selected_date = request.args.get('date') or get_kst_now().strftime('%Y-%m-%d') search_keyword = request.args.get('search', '') conn = sqlite3.connect(DB_PATH) conn.row_factory = sqlite3.Row c = conn.cursor() if search_keyword: c.execute("SELECT rowid, * FROM visitors WHERE visit_date = ? AND (name LIKE ? OR car_number LIKE ?) ORDER BY visit_time ASC", (selected_date, f"%{search_keyword}%", f"%{search_keyword}%")) else: c.execute("SELECT rowid, * FROM visitors WHERE visit_date = ? ORDER BY visit_time ASC", (selected_date,)) visitor_list = c.fetchall() conn.close() return render_template_string(desk_html, visitors=visitor_list, selected_date=selected_date, search_keyword=search_keyword, visitor_count=len(visitor_list)) @app.route('/delete_visitor', methods=['POST']) def delete_visitor(): if not session.get('logged_in'): return redirect('/login') rowid = request.form.get('rowid') conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute("DELETE FROM visitors WHERE rowid = ?", (rowid,)) conn.commit() conn.close() return redirect(f'/desk?date={request.form.get("date")}&search={request.form.get("search")}') # 🚀 5. 카메라가 입차를 찍는 시뮬레이션 @app.route('/simulate_entry', methods=['POST']) def simulate_entry(): if not session.get('logged_in'): return redirect('/login') rowid = request.form.get("rowid") now_time = get_kst_now().strftime('%H:%M') conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute("UPDATE visitors SET entry_time = ? WHERE rowid = ?", (now_time, rowid)) conn.commit() conn.close() return redirect(f'/desk?date={request.form.get("date")}&search={request.form.get("search")}') # 🚀 6. 카메라가 출차를 찍는 시뮬레이션 (새로 추가됨!) @app.route('/simulate_exit', methods=['POST']) def simulate_exit(): if not session.get('logged_in'): return redirect('/login') rowid = request.form.get("rowid") now_time = get_kst_now().strftime('%H:%M') conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute("UPDATE visitors SET exit_time = ? WHERE rowid = ?", (now_time, rowid)) conn.commit() conn.close() return redirect(f'/desk?date={request.form.get("date")}&search={request.form.get("search")}') @app.route('/approve', methods=['POST']) def approve(): if not session.get('logged_in'): return redirect('/login') rowid = request.form.get("rowid") added_time = request.form.get("parking_time") conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute("SELECT parking_time FROM visitors WHERE rowid = ?", (rowid,)) result = c.fetchone() current_time = result[0] if result and result[0] else "" new_time = added_time if current_time == "종일권" or added_time == "종일권": new_time = "종일권" elif current_time: try: curr_h = int(current_time.replace('시간', '')) add_h = int(added_time.replace('시간', '')) new_time = f"{curr_h + add_h}시간" except: new_time = added_time c.execute("UPDATE visitors SET status = '승인완료', parking_time = ? WHERE rowid = ?", (new_time, rowid)) conn.commit() conn.close() return redirect(f'/desk?date={request.form.get("date")}&search={request.form.get("search")}') @app.route('/download') def download(): if not session.get('logged_in'): return redirect('/login') selected_date = request.args.get('date') or get_kst_now().strftime('%Y-%m-%d') search_keyword = request.args.get('search', '') conn = sqlite3.connect(DB_PATH) c = conn.cursor() # 엑셀 다운로드 쿼리에도 purpose(목적), exit_time(출차시간) 추가 if search_keyword: c.execute("SELECT visit_time, entry_time, exit_time, name, visitor_company, car_number, host_info, purpose, status, parking_time FROM visitors WHERE visit_date = ? AND (name LIKE ? OR car_number LIKE ?) ORDER BY visit_time ASC", (selected_date, f"%{search_keyword}%", f"%{search_keyword}%")) else: c.execute("SELECT visit_time, entry_time, exit_time, name, visitor_company, car_number, host_info, purpose, status, parking_time FROM visitors WHERE visit_date = ? ORDER BY visit_time ASC", (selected_date,)) rows = c.fetchall() conn.close() si = StringIO() si.write('\ufeff') cw = csv.writer(si) cw.writerow(['방문 예정 시간', '입차 시간', '출차 시간', '방문자 성함', '소속', '차량 번호', '방문 대상', '방문 목적', '상태', '주차시간(누적)']) cw.writerows(rows) return Response(si.getvalue(), mimetype="text/csv", headers={"Content-Disposition": f"attachment;filename=visitors_{selected_date}.csv"})