Esempio n. 1
0
        print(Exception)
        sys.exit()

    # Build Query
    query_selectors = Query(**var_args).build_query()
    #print("Query selectors",query_selectors)

    # Get Results
    try:
        results = NEOSearcher(db).get_objects(query_selectors)

    except UnsupportedFeature as e:
        print('Unsupported Feature; Write unsuccessful')
        sys.exit()

    # Output Results
    try:
        result = NEOWriter().write(
            data=results,
            format=args.output,
        )
    except Exception as e:
        print(e)
        print('Write unsuccessful')
        sys.exit()

    if result:
        print('Write successful.')
    else:
        print('Write unsuccessful.')
Esempio n. 2
0
        sys.exit()
    except Exception as e:
        print(Exception)
        sys.exit()

    # Build Query
    query_selectors = Query(**var_args).build_query()

    # Get Results
    try:
        results = NEOSearcher(db).get_objects(query_selectors)
    except UnsupportedFeature as e:
        print('Unsupported Feature; Write unsuccessful')
        sys.exit()

    # Output Results
    try:
        result = NEOWriter().write(data=results,
                                   format=args.output,
                                   return_object=query_selectors.return_object)

    except Exception as e:
        print(e)
        print('Write unsuccessful')
        sys.exit()

    if result:
        print('Write successful.')
    else:
        print('Write unsuccessful.')
from search import *
# Testing purpose file

PROJECT_ROOT = pathlib.Path(__file__).parent.absolute()
input_filename = f'{PROJECT_ROOT}/data/neo_data.csv'
output_filename = f'{PROJECT_ROOT}/data/neo_data_out.csv'

# First step, load database
db = NEODatabase(input_filename)
db.load_data()

# getting the NEOs with 8 orbits
# obj = []
# for key in db.NEOList:
#     print(db[key])
# if len(db.NEOList[key].orbits) == 8:
#     obj.append(db.NEOList[key])

# "distance:>:74768000"
query_selectors = Query(**{
    "output": "csv_file",
    "start_date": "2020-01-01",
    "end_date": "2020-01-10",
    # "date": "2020-01-02",
    "number": 10,
    "filter": ["is_hazardous:=:False"]
}).build_query()

results = NEOSearcher(db).get_objects(query_selectors)
(NEOWriter().write("display", results, db))
            f'File {var_args.get("filename")} not found, please try another file name.'
        )
        sys.exit()
    except Exception as e:
        print(Exception)
        sys.exit()

    # Build Query
    query_selectors = Query(**var_args).build_query()

    # Get Results
    try:
        results = NEOSearcher(db).get_objects(query_selectors)
    except UnsupportedFeature as e:
        print('Unsupported Feature; Write unsuccessful')
        sys.exit()

    # Output Results
    try:
        output_filename = f'{PROJECT_ROOT}/data/neo_data_out.csv'
        result = NEOWriter().write(args.output, results, db)
    except Exception as e:
        print(e)
        print('Write unsuccessful')
        sys.exit()

    if result:
        print('Write successful.')
    else:
        print('Write unsuccessful.')