adding import command

This commit is contained in:
lza_menace 2020-10-21 22:50:40 -07:00
parent 157056ae9a
commit 62b43ca2b9
1 changed files with 15 additions and 0 deletions

View File

@ -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")