summaryrefslogtreecommitdiff
path: root/cli/app/commands/biggan/fetch.py
blob: 9d679f9950b5ea73e27eb895a68cd94b903e8641 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import click
import os

from app.utils.cortex_utils import download_cortex_files, find_unprocessed_files

@click.command('')
@click.option('-i', '--folder_id', 'opt_folder_id', type=int,
  help='Folder ID to fetch')
@click.pass_context
def cli(ctx, opt_folder_id):
  """
  Fetch new images from the server
  """
  files = download_cortex_files(opt_folder_id)
  unprocessed_files = find_unprocessed_files(files)
  print("Unprocessed files:")
  for file in unprocessed_files:
    print(" - {}".format(file['name']))