support new userdata.csv format

This commit is contained in:
Lane Jennison
2020-07-26 10:09:01 -04:00
parent 8187a5c08d
commit c805d3bfef
2 changed files with 7 additions and 6 deletions

View File

@@ -18,18 +18,19 @@ class Parser():
def transform_csv_data(self, data):
transformed = {}
key = ""
row_counter = 0
for line in data:
try:
key = '/'.join(line[0:2])
transformed[key] = '/'.join(line[2:5])
row_counter = row_counter + 1
key = line[0]
transformed[key] = line[1]
except IndexError as e:
print("Missing fields in data '{0}'. Error: '{1}'".format(line, e))
break
print("userdata.csv line {}: Malformed or Missing fields in data '{}'. Error: '{}'".format(row_counter, line, e))
return transformed
def import_csv(self):
with open(self.csv_file) as f:
csvreader = csv.reader(f, delimiter='/')
csvreader = csv.reader(f, delimiter='|')
data = list(csvreader)
return data

View File

@@ -21,7 +21,7 @@ app = Flask(__name__)
def get_redirect(path, IP):
split_path = path.split('/')
if len(split_path) == 2:
key = "{}/{}".format(split_path[0], split_path[1])
key = "/{}/{}".format(split_path[0], split_path[1])
new_path = dl_map.get(key, path)
return "{}/{}".format(mirror.next(), new_path)
if path == '':