mirror of
https://github.com/fastogt/fastocloud_admin.git
synced 2025-03-09 23:38:52 +00:00
34 lines
1.1 KiB
Python
Executable file
34 lines
1.1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import argparse
|
|
import os
|
|
import sys
|
|
from mongoengine import connect
|
|
|
|
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
|
|
|
|
from app.common.stream.entry import TestLifeStream
|
|
from app.service.service import ServiceSettings
|
|
from app.common.utils.m3u_parser import M3uParser
|
|
|
|
PROJECT_NAME = 'test_life'
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(prog=PROJECT_NAME, usage='%(prog)s [options]')
|
|
parser.add_argument('uri', help='Uri to m3u8 list')
|
|
parser.add_argument('mongo_uri', help='MongoDB credentials')
|
|
|
|
argv = parser.parse_args()
|
|
|
|
mongo = connect(argv.mongo_uri)
|
|
if mongo:
|
|
service_settings = ServiceSettings.objects().first()
|
|
m3u_parser = M3uParser()
|
|
m3u_parser.read_m3u(argv.uri)
|
|
m3u_parser.parse()
|
|
for file in m3u_parser.files:
|
|
stream = TestLifeStream.make_stream(service_settings)
|
|
stream.input.urls[0].uri = file['link']
|
|
stream.name = '{0}({1})'.format(file['tvg-group'], file['title'])
|
|
service_settings.streams.append(stream)
|
|
|
|
service_settings.save()
|