62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import os
|
||
|
|
import requests
|
||
|
|
|
||
|
|
import logging
|
||
|
|
from logging.handlers import RotatingFileHandler
|
||
|
|
|
||
|
|
def setup_logger(logger_name, log_file, backup=5):
|
||
|
|
logger = logging.getLogger(logger_name)
|
||
|
|
fmt = '%(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)s'
|
||
|
|
format_str = logging.Formatter(fmt)
|
||
|
|
file_handler = RotatingFileHandler(log_file, mode='a', maxBytes=500 * 1024 * 1024,
|
||
|
|
backupCount=backup, encoding='utf-8')
|
||
|
|
file_handler.setFormatter(format_str)
|
||
|
|
logger.addHandler(file_handler)
|
||
|
|
logger.setLevel(logging.INFO)
|
||
|
|
return logger
|
||
|
|
|
||
|
|
log_file_path = './roop.log'
|
||
|
|
logger = setup_logger('roop_logger', log_file_path)
|
||
|
|
|
||
|
|
from roop import core
|
||
|
|
|
||
|
|
def init_run_env():
|
||
|
|
path = os.getenv('path')
|
||
|
|
print(path)
|
||
|
|
current_path = os.path.abspath(os.curdir)
|
||
|
|
ffmpeg_path = os.path.join(current_path, 'ffmpeg')
|
||
|
|
path += ffmpeg_path + ';'
|
||
|
|
print(path)
|
||
|
|
os.environ['path'] = path
|
||
|
|
print(os.getenv('path'))
|
||
|
|
|
||
|
|
|
||
|
|
def upload_post(paper, download_url):
|
||
|
|
http_server_ip = '192.168.43.242'
|
||
|
|
url = 'http://{}:5000/upload'.format(http_server_ip)
|
||
|
|
print('upload file:', url)
|
||
|
|
with open(paper, 'rb') as file:
|
||
|
|
files = {'photo': file} # 构建包含图片文件的字典
|
||
|
|
data = {'url': download_url}
|
||
|
|
response = requests.post(url, files=files, data=data) # 发送 POST 请求进行图片上传
|
||
|
|
print('response:', response.content, response.status_code)
|
||
|
|
if response.status_code == 200:
|
||
|
|
print('图片上传成功')
|
||
|
|
else:
|
||
|
|
print('图片上传失败')
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
# httpserver.start_http_server()
|
||
|
|
# player = VideoPlayer('./movie/video.mp4')
|
||
|
|
|
||
|
|
init_run_env()
|
||
|
|
core.run()
|
||
|
|
# url = 'http://127.0.0.1:7001/public/upload/poster/1695864631144_42fb302c-9c87-4b2c-9a89-d8f7e60f5346_2023_09_28_093021.jpg_peper.png'
|
||
|
|
# upload_post('./saved/out/2023_09_28_115912.jpg_peper.png', url)
|
||
|
|
logger.info('exit')
|
||
|
|
# httpserver.stop_http_server()
|
||
|
|
|