这次更新主要是更新文件自动入库时，同时也向数据库写入文件的时长。例如一个视频长度为01:25:30:02，需要将该数据从视频中提取出来并保存至数据库，同时在查看视频详情时可以展示。
修改数据表，新增durtaion字段
models.py
#媒体信息列表
class content_media(models.Model):
id = models.AutoField(primary_key=True)
create_time=models.DateTimeField()
name=models.CharField(max_length=40)
size=models.CharField(max_length=15)
path=models.CharField(max_length=200)
duration=models.CharField(max_length=12,default='00:00:00:00')
在命令行中执行：
python manage.py makemigrations
python manage.py migrate
数据表更新完成
修改主程序代码如下：
import os
import pymysql
import time
import copy
import os
import cv2
def scan_files(directory):
files_list = []
for root, sub_dirs, files in os.walk(directory):
for special_file in files:
tmp = []
if (special_file.split('_')[0] != 'aizou0629'):
#判断文件完全写入才进行后续操作
if (isFinished(os.path.join(root, special_file))):
#判断文件是否出现重名
if ('aizou0629_' + special_file in files):
tmpname = copy.deepcopy(special_file)
flag=1
while 1:
if('aizou0629_' +special_file.replace('.', '('+str(flag)+').') not in files):
special_file=special_file.replace('.', '('+str(flag)+').')
os.rename(os.path.join(root, tmpname), os.path.join(root, special_file))
break
else:
flag+=1
#开始搜集需入库的文件列表
tmp.append(special_file)
tmp.append(cal_size(os.path.join(root, special_file)))
timearray = time.localtime(os.path.getctime(os.path.join(root, special_file)))
tmp.append(time.strftime("%Y-%m-%d %H:%M:%S", timearray))
tmp.append(getMediaInfo(os.path.join(root, special_file)))
files_list.append(tmp)
if (len(files_list) == 0):
return 0
conn = pymysql.connect(host='localhost',
user='root',
password='ChenzroyI.*?',
database='test')
cursor = conn.cursor()
for each in files_list:
path = str(os.path.join(root, 'aizou0629_' + each[0]))
NewPath = ""
for each_1 in path.split("\\"):
NewPath += each_1 + "\\\\"
NewPath = NewPath[:len(NewPath) - 2]
sql="insert into myapp_content_media (create_time,name,size,path,duration) values ('{}','{}','{}','{}','{}')".format(each[2],each[0],each[1],NewPath,each[3])
cursor.execute(sql)
file_rename(root, each[0])
conn.commit()
cursor.close()
conn.close()
return files_list
修改文件大小格式，使其更符合阅读习惯
def cal_size(path):
flag = 1
unit_dict = {1: 'B', 2: 'KB', 3: 'MB', 4: 'GB', 5: 'TB'}
size = os.path.getsize(path)
while 1:
if (size > 1024):
size = size / 1024
flag += 1
else:
return str(round(size, 2)) + unit_dict[flag]
判断外来文件是否已完全拷贝到存储中
def isFinished(path):
tmp = []
while 1:
size = os.path.getsize(path)
if (size in tmp):
tmp.append(size)
else:
del tmp
tmp = [size]
if (len(tmp) == 3):
return True
time.sleep(10)
def file_rename(root, filename):
filename_new = 'aizou0629_' + filename
os.rename(os.path.join(root, filename), os.path.join(root, filename_new))
#获取视频的时长
def getMediaInfo(path):
cap = cv2.VideoCapture(path) #打开视频文件
n_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) #视频的帧数
fps = cap.get(cv2.CAP_PROP_FPS) #视频的帧率
media_f = int(n_frames % fps) #视频时长的帧部分
media_s = int(n_frames // fps % 60) #视频时长的秒部分
media_m = int(n_frames // fps // 60 % 60) #视频时长的分钟部分
media_h = int(n_frames // fps // 60 // 60) #视频时长的小时部分
def int2str(media_x):
media_x=str(media_x)
if(len(media_x)==1):
return '0'+media_x
else:
return media_x
media_f=int2str(media_f)
media_s=int2str(media_s)
media_m=int2str(media_m)
media_h=int2str(media_h)
return media_h+':'+media_m+':'+media_s+':'+media_f
if __name__ == "__main__":
path = r'D:\PythonWorkspace\MyFirstDjango\mysite\content_media\origin'
while 1:
scan_files(path)
time.sleep(5)
本次更新中新增加了一个getMediaInfo的方法，该方法主要用来提取视频的时长。
如果直接通过cv2来提取视频时长的话只能获得视频的帧数，比如一个3分钟FPS为25帧的视频，我们想要的是这种格式：’00:03:00:00’，但是用cv2提取出来的则是：’4500’，也就是视频的总帧数。
通过getMediaInfo方法可以直接将其转换为我们想要的格式。
Original: https://blog.csdn.net/weixin_42144330/article/details/123785314
Author: 跟着皮皮实战Python
Title: 从零开始打造云端AI管理调度平台（番外篇_1）文件自动入库部分的更新
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/736037/
转载文章受原作者版权保护。转载请注明原作者出处！