[Python Flask] - 파일 서버 만들기

위와 같이 간단한 파일 서버를 구현했습니다. 파일 이름을 클릭하면 다운로드되고 삭제를 누르면 파일이 삭제됩니다. 파일을 선택하고 제출을 누르면 파일이 업로드됩니다. 코드는 아래와 같습니다. 플라스크와 신사2 템플릿 그리고 자바스크립트를 사용하여 코딩했습니다.

app.py

from flask import Flask

from flask import render_template

from flask import request

from flask import redirect, url_for

from flask import send_file # 파일 다운로드를 위한 모듈

from flask_wtf import FlaskForm # 플라스크 폼

from flask_wtf.file import FileField, FileRequired # FileField -> 파일 입력하는 필드, FileRequired -> 파일 유효성 검사

import os

import datetime

import time

app = Flask(__name__)

app.config['SECRET_KEY'] = "secret" # CSRF 공격 방지 토큰

class FileForm(FlaskForm): # 파일 유효성 검사하는 폼 클래스 생성

files = FileField(validators=[FileRequired('업로드할 파일을 넣어주세요')])

def stamp2real(stamp):

return datetime.datetime.fromtimestamp(stamp)

def info(filename):

ctime = os.path.getctime(filename) # 만든시간

mtime = os.path.getmtime(filename) # 수정시간

size = os.path.getsize(filename) # 파일크기 (단위: bytes)

return ctime, mtime, size

@app.errorhandler(404) # 404에러 처리

def page_not_found(error):

return render_template('deny.html', pwd=os.getcwd() + "\\uploads"), 404

@app.route('/', methods = ['GET', 'POST'])

def main_page():

form = FileForm() # 파일 유효성 폼 클래스 인스턴스 생성

if form.validate_on_submit(): # 양식 유효성 검사 + POST인 경우

f = form.files.data

f.save('./uploads/' + f.filename) # 파일 업로드

return render_template('check.html', pwd=os.getcwd() + "\\uploads")

filelist = os.listdir('./uploads') # 파일 리스트 가져오기

infos = []

for name in filelist:

fileinfo = {}

ctime, mtime, size = info('./uploads/' + name)

fileinfo["name"] = name

fileinfo["create"] = stamp2real(ctime)

fileinfo["modify"] = stamp2real(mtime)

if(size <= 1000000):

fileinfo["size"] = "%.2f KB" % (size / 1024)

else:

fileinfo["size"] = "%.2f MB" % (size / (1024.0 * 1024.0))

infos.append(fileinfo) # 각 파일 정보를 모두 리스트로 입력

return render_template('home.html', form=form,

pwd=os.getcwd() + "\\uploads", infos = infos)

@app.route('/down/<path:filename>')

def down_page(filename):

return send_file('uploads/' + filename, # 다운로드할 파일 uploads/ 경로에서 가져와서 반환

attachment_filename = filename,

as_attachment=True)

@app.route('/del/<path:filename>')

def delete_page(filename):

os.remove('uploads/'+ filename) # 파일 삭제

return "<script>location.href = \"/\";</script>" # 루트 페이지로 이동

if __name__ == '__main__':

#서버 실행

app.run()

home.html

{% extends "base.html" %}

{% block content %}

<body>

{{ form.csrf_token() }}

{{ form.files(size=20) }}

</form>

</div>

<br>

<thead>

<tr>

<th>마지막으로 수정한 날짜</th>

</tr>

</thead>

<tbody>

{% for info in infos %}

<tr>

{% if info['name'].split('.')[1].lower() == "png"

or info['name'].split('.')[1].lower() == "jpg"

or info['name'].split('.')[1].lower() == "jpeg"

or info['name'].split('.')[1].lower() == "gif" %}

<td class="badge badge-pill badge-success align-middle">{{info['name'].split(".")[1].lower()}}</td> # 초록색 확장자 이미지

{% elif info['name'].split('.')[1].lower() == "7z"

or info['name'].split('.')[1].lower() == "zip" %}

<td class="badge badge-pill badge-primary align-middle">{{info['name'].split(".")[1].lower()}}</td>

{% else %}

<td class="badge badge-pill badge-light align-middle">{{info['name'].split(".")[1].lower()}}</td>

{% endif %}

<td>{{ info['create'].strftime("%Y-%m-%d") }}</td> # 만든 날짜 출력

<td>{{ info['modify'].strftime("%Y-%m-%d") }}</td> # 수정 날짜 출력

</tr>

{% endfor %}

</tbody>

</table>

</div>

</body>

{% endblock %}

base.html

<!DOCTYPE html>

<head>

<title>Server</title>

</head>

<body>

<h3><i class="fa fa-folder-open fa-lg"></i> FileServer</h3>

</div>

</span>

</div>

{% block content %}{% endblock %}

</body>

</html>

check.html

{% extends "base.html" %}

{% block content %}

<br>

</div>

function onload(){

setTimeout('go_home()',3000)

}

function go_home(){

location.href="/"

}

</script>

</body>

{% endblock %}

deny.html

{% extends "base.html" %}

{% block content %}

<br>

<h2>잘못 접근하셨어요 :(</h2>

</div>

function onload(){

setTimeout('go_home()',5000)

}

function go_home(){

location.href="/"

}

</script>

</body>

{% endblock %}

저작자표시

'웹 개발 > Back End' 카테고리의 다른 글

스프링 입문 - 템플릿 엔진 개념 (0)	2021.12.31
파이썬 Flask와 PostgreSQL를 이용한 쇼핑몰 구현 프로젝트 (0)	2021.12.15
[Python Flask] - Todo 메모장 만들기 (0)	2021.11.14
파이썬 플라스크(Flask)의 Jinja2(신사2)를 이용한 구구단 구현 (0)	2021.11.12
웹 프레임워크 및 IDE(통합개발환경) 종류 (0)	2021.11.09

L3m0n $0ju

[Python Flask] - 파일 서버 만들기

'웹 개발 > Back End' 카테고리의 다른 글

댓글

티스토리툴바

[Python Flask] - 파일 서버 만들기

'웹 개발 > Back End' 카테고리의 다른 글

관련글

댓글

티스토리툴바