python 截屏录屏

gaohaikuo1

直接运行版

1. import pyautogui
   
2. import cv2
   
3. import numpy as np
   
4. import datetime
   
5. import time
   
6. def get_datetime_str(style='dt'):
   
7.     cur_time = datetime.datetime.now()
   
8. 
   
9.     date_str = cur_time.strftime('%y%m%d')
   
10.     time_str = cur_time.strftime('%H%M%S')
    
11. 
    
12.     if style == 'data':
    
13.         return date_str
    
14.     elif style == 'time':
    
15.         return time_str
    
16.     else:
    
17.         return date_str + '_' + time_str
    
18. 
    
19. path = "D:/image/"
    
20. 
    
21. num_frame = 200
    
22. vw = cv2.VideoWriter(path+get_datetime_str()+'video.avi', cv2.VideoWriter_fourcc('I','4','2','0'), 8,(800,450))
    
23. while num_frame:
    
24.     print(num_frame)
    
25.     num_frame = num_frame-1
    
26.     #img = pyautogui.screenshot(region=[300,50, 500, 500])
    
27.     img = pyautogui.screenshot()
    
28.     # 分别代表：左上角坐标，宽高
    
29.     img=img.resize((800,450))
    
30. #   img.save(path+get_datetime_str()+".jpg")
    
31.     #转成视频
    
32.     #对获取的图片转换成二维矩阵形式，后再将RGB转成BGR
    
33.     #因为imshow,默认通道顺序是BGR，而pyautogui默认是RGB所以要转换一下，不然会有点问题
    
34.     img = cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)   
    
35.     vw.write(img)
    
36.     time.sleep(1);
    
37. vw.release()
    
38. #cv2.imshow("截屏",img)
    
39. 
    
40. #cv2.waitKey(0)
    
41. 
    

复制代码

gaohaikuo1

主界面

1. from tkinter import *
   
2. import pyautogui
   
3. import cv2
   
4. import numpy as np
   
5. import datetime
   
6. import time
   
7. from screenObj import *
   
8. top = Tk()
   
9. top.title("考核实况记录表")
   
10. top.geometry('300x100')
    
11. 
    
12. txt = StringVar()
    
13. b1=Entry(top,width=20)
    
14. b1.pack()
    
15. b3=Entry(top,width=20, textvariable=txt)
    
16. b3.pack()
    
17. 
    
18. sco = screenObj(txt)
    
19. def start_screen():
    
20.     sco.setText(b1.get())
    
21.     print(b1.get())
    
22.     sco.start_screen()
    
23.     pass
    
24. 
    
25. 
    
26. def stop_screen():
    
27.     sco.stop_screen()
    
28.     txt.set('录制停止')
    
29.     pass
    
30. #添加按钮1
    
31. first = Button(top, text="开始录制", fg="tomato", command=start_screen)
    
32. first.pack()
    
33. #添加按钮2
    
34. second = Button(top, text="停止录制", fg="tomato", command=stop_screen)
    
35. second.pack()
    
36. top.mainloop()

复制代码

录屏程序

1. import pyautogui
   
2. import cv2
   
3. import numpy as np
   
4. import datetime
   
5. import time
   
6. import threading
   
7. def get_datetime_str(style='dt'):
   
8.     cur_time = datetime.datetime.now()
   
9. 
   
10.     date_str = cur_time.strftime('%y%m%d')
    
11.     time_str = cur_time.strftime('%H%M%S')
    
12. 
    
13.     if style == 'data':
    
14.         return date_str
    
15.     elif style == 'time':
    
16.         return time_str
    
17.     else:
    
18.         return date_str + '_' + time_str
    
19. class screenObj():
    
20.     def __init__(self,txt = '测试'):
    
21.         self.path = "D:/image/"      
    
22.         self.txt  = txt
    
23.         
    
24.     def setText(self,mark):        
    
25.         self.mark  = mark
    
26.         
    
27.     def start_screen(self):
    
28.         self.vw = cv2.VideoWriter(self.path+self.mark+get_datetime_str()+'.avi',
    
29.                                   cv2.VideoWriter_fourcc('I','4','2','0'), 8,(800,450))
    
30.         self.on_screen = True
    
31.         threading.Thread(target = self.screen_loop).start()
    
32.         
    
33.     def stop_screen(self):
    
34.         self.on_screen = False
    
35.         
    
36.     def screen_loop(self):
    
37.         self.frame = 1
    
38.         while self.on_screen:
    
39.             print("帧数"+str(self.frame))
    
40.             self.txt.set('已经录制'+str(self.frame)+'秒')
    
41.             self.frame = self.frame + 1
    
42.             img = pyautogui.screenshot()
    
43.             # 分别代表：左上角坐标，宽高
    
44.             img=img.resize((800,450))
    
45.             img = cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)
    
46.             cv2.putText(img,text= self.mark ,org=(40, 90),
    
47.                         fontFace=cv2.FONT_HERSHEY_SIMPLEX,fontScale= 2,
    
48.                         color=(255,0,0),thickness=10,lineType=cv2.LINE_4)
    
49. 
    
50.             self.vw.write(img)
    
51.             time.sleep(1);
    
52.         self.vw.release()
    
53.         print("录制结束")
    
54.    
    
55. if __name__ == '__main__':
    
56.     sco = screenObj()
    
57.     sco.setText('aaa')
    
58.     sco.start_screen()
    
59.     time.sleep(5)
    
60.     sco.stop_screen()
    
61.    
    
62. 
    

复制代码