从零开始打造云端AI管理调度平台（番外篇1）文件自动入库更新

　　这次更新主要是更新文件自动入库时，同时也向数据库写入文件的时长。例如一个视频长度为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:PythonWorkspaceMyFirstDjangomysitecontent_mediaorigin＂     while 1:         scan_files(path)           time.sleep(5)
　　本次更新中新增加了一个getMediaInfo的方法，该方法主要用来提取视频的时长。
　　如果直接通过cv2来提取视频时长的话只能获得视频的帧数，比如一个3分钟FPS为25帧的视频，我们想要的是这种格式：＂00:03:00:00＂，但是用cv2提取出来的则是：＂4500＂，也就是视频的总帧数。
　　通过getMediaInfo方法可以直接将其转换为我们想要的格式。
　　如果文章帮助到了您，可否给一个点赞关注收藏呢~
　　如果遇到编程上的问题，欢迎留言哦~
　　关注作者回复：aiweb，获取源码下载地址哦~