24 lines
623 B
Python
24 lines
623 B
Python
from django.db import models
|
|
import PyPDF2
|
|
|
|
|
|
class PDFDocument(models.Model):
|
|
title = models.CharField(max_length=255)
|
|
file = models.FileField(upload_to='pdfs/')
|
|
content = models.TextField(blank=True)
|
|
uploaded_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
def save(self, *args, **kwargs):
|
|
if self.file:
|
|
pdf = PyPDF2.PdfReader(self.file)
|
|
texto = []
|
|
for pagina in pdf.pages:
|
|
texto.append(pagina.extract_text())
|
|
self.content = '\n'.join(texto)
|
|
|
|
super().save(*args, **kwargs)
|
|
|