mirror of
https://github.com/armbian/dl-router.git
synced 2026-01-06 10:32:39 -08:00
support new userdata.csv format
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
@@ -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 == '':
|
||||
|
||||
Reference in New Issue
Block a user