diff --git a/scripts/import_subscribers_from_xtream.py b/scripts/import_subscribers_from_xtream.py new file mode 100755 index 0000000..b106fc5 --- /dev/null +++ b/scripts/import_subscribers_from_xtream.py @@ -0,0 +1,38 @@ +#!/usr/bin/env python3 +import argparse +import os +import sys +from mongoengine import connect +import mysql.connector + +sys.path.append(os.path.join(os.path.dirname(__file__), '..')) + +from app.common.subscriber.login.entry import SubscriberUser + +PROJECT_NAME = 'create_provider' + +if __name__ == '__main__': + parser = argparse.ArgumentParser(prog=PROJECT_NAME, usage='%(prog)s [options]') + parser.add_argument('--mongo_uri', help='MongoDB credentials', default='mongodb://localhost:27017/iptv') + parser.add_argument('--mysql_host', help='MySQL host', default='localhost') + parser.add_argument('--mysql_user', help='MySQL password', default='') + parser.add_argument('--mysql_password', help='MySQL password', default='') + + argv = parser.parse_args() + mysql_host = argv.mysql_host + mysql_user = argv.mysql_user + mysql_password = argv.mysql_password + + mongo = connect(host=argv.mongo_uri) + if not mongo: + sys.exit(1) + + mydb = mysql.connector.connect( + host=mysql_host, + user=mysql_user, + passwd=mysql_user + ) + + for sql_entry in table: + new_user = SubscriberUser.make_subscriber(email=sql_entry.username, password=sql_entry.password, country='US') + new_user.save()