adding import command

i2p
lza_menace 4 years ago
parent 157056ae9a
commit 62b43ca2b9

@ -3,6 +3,7 @@ import json
import requests import requests
import re import re
import logging import logging
import click
from os import makedirs from os import makedirs
from datetime import datetime from datetime import datetime
from flask import Flask, request, redirect from flask import Flask, request, redirect
@ -159,11 +160,25 @@ def export():
export_dir = f"{config.DATA_DIR}/export.txt" export_dir = f"{config.DATA_DIR}/export.txt"
nodes = Node.select().where(Node.validated == True) nodes = Node.select().where(Node.validated == True)
for node in nodes: for node in nodes:
logging.info(f"Adding {node.url}")
all_nodes.append(node.url) all_nodes.append(node.url)
with open(export_dir, "w") as f: with open(export_dir, "w") as f:
f.write("\n".join(all_nodes)) f.write("\n".join(all_nodes))
logging.info(f"{nodes.count()} nodes written to {export_dir}") logging.info(f"{nodes.count()} nodes written to {export_dir}")
@app.cli.command("import")
def import_():
all_nodes = []
export_dir = f"{config.DATA_DIR}/export.txt"
with open(export_dir, 'r') as f:
for url in f.readlines():
n = url.rstrip()
all_nodes.append(n)
logging.info(f"Adding {n}")
node = Node(url=n)
node.save()
logging.info(f"{len(all_nodes)} node urls imported and ready to be validated")
@app.template_filter("humanize") @app.template_filter("humanize")
def humanize(d): def humanize(d):
t = arrow.get(d, "UTC") t = arrow.get(d, "UTC")

Loading…
Cancel
Save