From cdb5ecc9a02eff13914cad39784163a53b79d56a Mon Sep 17 00:00:00 2001
From: Gui Heise <gheise@ufl.edu>
Date: Wed, 19 Jan 2022 16:59:25 -0500
Subject: [PATCH] Fix datetime support

---
 cli.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cli.py b/cli.py
index e771252..e480ea8 100644
--- a/cli.py
+++ b/cli.py
@@ -119,8 +119,8 @@ def fetch_all_prices():
 
 
 @cli.command()
-@click.argument("after", type=datetime)
-@click.argument("before", type=datetime)
+@click.argument("after", type=click.DateTime(formats=["%Y-%m-%d", "%m-%d-%Y"]))
+@click.argument("before", type=click.DateTime(formats=["%Y-%m-%d", "%m-%d-%Y"]))
 def fetch_range(after: datetime, before: datetime):
     inspect_db_session = get_inspect_session()