s426206/WTV2D
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add video upload

Browse Source
This commit is contained in:
Jan Nowak 2022-06-09 15:11:48 +02:00
parent 1712349965
commit f82275df71
7 changed files with 108 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
WTV2D/settings.py
View File

@ -117,5 +117,8 @@ USE_TZ = True
# Static files (CSS, JavaScript, Images)
# https://docs.djangoproject.com/en/3.0/howto/static-files/
STATIC_ROOT = os.path.join(BASE_DIR, 'static/')
STATIC_URL = '/static/'
MEDIA_ROOT = os.path.join(BASE_DIR, 'media') #folder przechowywanych zdjec
MEDIA_URL = '/media/'

9
WTV2D/urls.py
View File

@ -15,8 +15,17 @@ Including another URLconf
"""
from django.contrib import admin
from django.urls import path, include
from django.conf import settings
from django.conf.urls.static import static
from django.contrib.staticfiles.urls import staticfiles_urlpatterns
urlpatterns = [
path('admin/', admin.site.urls),
path('', include('core.urls')),
]
if settings.DEBUG:
urlpatterns += static(settings.MEDIA_URL, document_root=settings.MEDIA_ROOT)
else:
urlpatterns += staticfiles_urlpatterns()

13
core/forms.py Normal file
View File

@ -0,0 +1,13 @@
from django import forms
from .models import VideoFile
# Form for upload video to the server
class VideoUploadForm(forms.ModelForm):
class Meta:
model = VideoFile
fields = ['file']
def __init__(self, *args, **kwargs):
super(VideoUploadForm, self).__init__(*args, **kwargs)
# In html input only accept video uploads
self.fields['file'].widget.attrs.update({'class': 'form-control border-0', 'accept': 'video/*'})

28
core/models.py
View File

@ -1,3 +1,31 @@
from django.db import models
from enum import Enum
# Create your models here.
class VideoFile(models.Model):
"""
A class for uploading video to the server.
"""
def forDjango(cls):
cls.do_not_call_in_templates = True
return cls
@forDjango
class VideoFileStatus(Enum):
NEW = 1
PROCESSING = 4
READY = 7
file = models.FileField(upload_to="", max_length=1024) # If upload_to=="" then file will be uploaded to the MEDIA_ROOT path set in settings.
uploaded_at = models.DateTimeField(auto_now_add=True)
fps = models.IntegerField(default=25)
video_width = models.IntegerField(default=0, blank=True)
video_height = models.IntegerField(default=0, blank=True)
status = models.IntegerField(default=VideoFileStatus.NEW.value)
@property
def url(self):
return self.file.url

0
core/migrations/templates/core/home.html → core/static/css/main.css
View File

31
core/templates/core/home.html
View File

@ -0,0 +1,31 @@
{% load static %}
<!DOCTYPE html>
<html>
<head>
<link href="https://cdn.jsdelivr.net/npm/bootstrap@5.0.2/dist/css/bootstrap.min.css" rel="stylesheet"
integrity="sha384-EVSTQN3/azprG1Anm3QDgpJLIm9Nao0Yz1ztcQTwFspd3yD65VohhpuuCOmLASjC" crossorigin="anonymous">
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
<script type="text/javascript" src="/static/js/jquery-3.6.0.min.js"></script>
<link rel="stylesheet" href="{% static 'css/main.css' %}">
<title>Web-TV2D</title>
</head>
<body>
<div>
<div class="d-flex flex-column min-vh-100 justify-content-center align-items-center">
<h1> Submit your video</h1>
<form id="form" method="POST" class="mt-4 text-center" enctype="multipart/form-data">
{% csrf_token %}
{% for message in messages %}
{% if message.level == DEFAULT_MESSAGE_LEVELS.ERROR %}
<span style="color: red">{{message}}</span>
{% elif message.level == DEFAULT_MESSAGE_LEVELS.SUCCESS %}
<span style="color: green">{{message}}</span>
{% endif %}
{% endfor %}
{{ form.file }}
<input type="submit" class="btn btn-primary mt-2" value="Upload">
</form>
</div>
</div>
</body>
</html>

24
core/views.py
View File

@ -1,7 +1,29 @@
import cv2
from django.http import HttpResponseRedirect
from django.shortcuts import render
from django.contrib import messages
from core.forms import VideoUploadForm
from core.models import VideoFile
# Create your views here.
def home(request):
return render(request, 'core/home.html')
form = VideoUploadForm()
if request.method == "POST":
form = VideoUploadForm(request.POST, request.FILES)
if form.is_valid():
video_file = VideoFile(file=request.FILES['file'])
video_file.save()
video = cv2.VideoCapture(video_file.file.path)
video_file.fps = video.get(cv2.CAP_PROP_FPS)
video_file.video_width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
video_file.video_height = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))
video_file.save()
else:
messages.error(request, "Something went wrong")
context = {'form': form}
return render(request, 'core/home.html', context)