快捷導(dǎo)航

Python如何將圖像音視頻等資源文件隱藏在代碼中(小技巧)

更新時(shí)間：2020年02月16日 11:23:20 作者：天元浪子

有朋友問(wèn)小編使用pyinstaller打包源碼時(shí)，因?yàn)榇a中使用了圖像、音頻、視頻等資源文件，無(wú)法將程序打包成單一的可執(zhí)行文件，怎么處理呢，下面小編通過(guò)代碼給大家介紹Python如何將圖像音視頻等資源文件隱藏在代碼中，感興趣的朋友一起看看吧

下午有同學(xué)Python學(xué)習(xí)群里說(shuō)，使用pyinstaller打包源碼時(shí)，因?yàn)榇a中使用了圖像、音頻、視頻等資源文件，無(wú)法將程序打包成單一的可執(zhí)行文件。有沒(méi)有方法將這些資源文件按保存在代碼中呢？我想了一下，應(yīng)該是可以的。于是乎，花了一個(gè)小時(shí)，寫出了下面的代碼，算是拋磚引玉吧。

這段代碼可以將二進(jìn)制文件轉(zhuǎn)存為python腳本文件，供其他腳本引用。代碼最后附有使用的例子，演示用的圖片可以隨便照一張。除了轉(zhuǎn)存二進(jìn)制數(shù)據(jù)，還提供了兩個(gè)方法：

get_fp()：返回二進(jìn)制的IO對(duì)象（類文件對(duì)象）
save()：保存為本地文件

# -*- coding: utf-8 -*-
"""以python模塊形式存儲(chǔ)、使用二進(jìn)制文件"""
import os
import base64
from io import BytesIO
def bin2module(bin_file, py_file=None):
  """二進(jìn)制文件轉(zhuǎn)存為python模塊
  bin_file  - 二進(jìn)制文件名
  py_file   - 生成的模塊文件名，默認(rèn)使用二進(jìn)制文件名，僅更改后綴名
  """
  fpath, fname = os.path.split(bin_file)
  fn, ext = os.path.splitext(fname)
  if not py_file:
    py_file = os.path.join(fpath, '%s.py'%fn)
  with open(bin_file, 'rb') as fp:
    content = fp.read()
  content = base64.b64encode(content)
  content = content.decode('utf8')
  with open(py_file, 'w') as fp:
    fp.write('# -*- coding: utf-8 -*-\n\n')
    fp.write('import base64\n')
    fp.write('from io import BytesIO\n\n')
    fp.write('content = """%s"""\n\n'%content)
    fp.write('def get_fp():\n')
    fp.write('  return BytesIO(base64.b64decode(content.encode("utf8")))\n\n')
    fp.write('def save(file_name):\n')
    fp.write('  with open(file_name, "wb") as fp:\n')
    fp.write('    fp.write(base64.b64decode(content.encode("utf8")))\n')
if __name__ == '__main__':
  """測(cè)試代碼"""
  # 將圖像文件轉(zhuǎn)存為img_demo.py
  bin2module('forever.png', 'demo.py')
  # 導(dǎo)入剛剛生成的demo模塊
  import demo
  # 用pillow打開(kāi)圖像，驗(yàn)證demo模塊的get_fp()：返回二進(jìn)制的IO對(duì)象（類文件對(duì)象）
  from PIL import Image
  im = Image.open(demo.get_fp())
  im.show()
  # 保存為本地文件，驗(yàn)證demo模塊的save()：保存文件
  demo.save('demo_save.png')

補(bǔ)充：下面看下Python實(shí)現(xiàn)將視頻按間隔截取為圖片（附代碼）

輸入：一段視頻。

輸出：取出的視頻幀。

準(zhǔn)備：新建一個(gè)文件夾，用來(lái)放置截出來(lái)視頻幀。

代碼實(shí)現(xiàn)：

import cv2
import argparse
import os
def parse_args():
  """
  Parse input arguments
  """
  parser = argparse.ArgumentParser(description='Process pic')
  parser.add_argument('--input', help='video to process', dest='input', default=None, type=str)
  parser.add_argument('--output', help='pic to store', dest='output', default=None, type=str)
  #default為間隔多少幀截取一張圖片
  parser.add_argument('--skip_frame', dest='skip_frame', help='skip number of video', default=100, type=int) #此處可更改提取幀的間隔
  args = parser.parse_args(['--input','','--output',''])  #此處添加路徑，input為輸入視頻的路徑 ，output為輸出存放圖片的路徑
  return args
def process_video(i_video, o_video, num):
  cap = cv2.VideoCapture(i_video)
  num_frame = cap.get(cv2.CAP_PROP_FRAME_COUNT)
  expand_name = '.jpg'
  if not cap.isOpened():
    print("Please check the path.")
  cnt = 0
  count = 0
  while 1:
    ret, frame = cap.read()
    cnt += 1
    # how
    # many
    # frame
    # to
    # cut
    if cnt % num == 0:
      count += 1
      cv2.imwrite(os.path.join(o_video, str(count) + expand_name), frame)
    if not ret:
      break
if __name__ == '__main__':
  args = parse_args()
  if not os.path.exists(args.output):
    os.makedirs(args.output)
  print('Called with args:')
  print(args)
  process_video(args.input, args.output, args.skip_frame)

運(yùn)行起來(lái)非常容易，若是出錯(cuò)請(qǐng)檢查路徑書寫是否正確。如下是一種絕對(duì)路徑的寫法舉例，前方加 r。