Add configuration file

This commit is contained in:
MisterDaneel 2019-02-16 18:00:51 +01:00
parent 363aafe901
commit 8316712c75
2 changed files with 36 additions and 5 deletions

View File

@ -16,8 +16,10 @@ logging.basicConfig(level=logging.DEBUG)
class torrent(): class torrent():
def __init__(self, torrent_tile): def __init__(self, configuration):
with open(torrent_tile, 'rb') as tf: self.configuration = configuration
torrent_file = self.configuration['torrent:']
with open(torrent_file, 'rb') as tf:
data = tf.read() data = tf.read()
self.b_enc = bencoding() self.b_enc = bencoding()
self.metainfo = self.b_enc.bdecode(data) self.metainfo = self.b_enc.bdecode(data)

35
main.py
View File

@ -1,6 +1,35 @@
from code.torrent import torrent from code.torrent import torrent
import argparse
import json
import sys
to = torrent('test.torrent') def parse_args():
rep = to.tracker_start_request() """Create the arguments"""
to.tracker_process() parser = argparse.ArgumentParser('\nratio.py -c <configuration-file.json>')
parser.add_argument("-c", "--configuration", help="Configuration file")
return parser.parse_args()
def load_configuration(configuration_file):
with open(configuration_file) as f:
configuration = json.load(f)
if 'torrent' not in configuration:
return None
return configuration
if __name__ == "__main__":
args = parse_args()
if args.configuration:
configuration = load_configuration(args.configuration)
else:
sys.exit()
if not configuration:
sys.exit()
to = torrent('test.torrent')
rep = to.tracker_start_request()
to.tracker_process()