initial submit

trphunt · trphunt · commit 461e445ffb31 · 2018-03-19T04:15:46.000-04:00
diff --git a/db.sqlite3 b/db.sqlite3
diff --git a/departures/migrations/0002_populate_departures.py b/departures/migrations/0002_populate_departures.py
@@ -0,0 +1,31 @@
+"""
+Loads data from `departures.json` into the Departure model
+"""
+
+import json
+from django.db import migrations, models
+
+def populate_departures(apps, schema_editor):
+    Departure = apps.get_model('departures', 'Departure')
+
+    with open('departures.json') as json_data:
+        data = json.load(json_data)
+        for d in data:
+            departure = Departure(
+                name=d['name'],
+                start_date=d['start_date'],
+                finish_date=d['finish_date'],
+                category=d['category']
+            )
+            departure.save()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('departures', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.RunPython(populate_departures),
+    ]
diff --git a/fetch_departures.py b/fetch_departures.py
@@ -0,0 +1,78 @@
+"""
+Solution for gadventures/python-api-challenge
+
+This module fetches departures data from an api server,
+filters by date and category, and then writes the filtered values
+to a csv file.
+
+Submitted by: Tyler Hunt
+Email: trphunt@tylerhunt.solutions
+Date: March 2018
+"""
+from datetime import datetime
+import csv
+import requests
+
+######## Fetch Data ########
+
+URL = 'http://localhost:8000/departures'
+HEADERS = {"Accept": "application/json"}
+
+def fetch_departures(output_list, url, headers):
+    """
+    Recursively loads pages of departures into list
+    until `next_url` is None.
+    """
+    response = requests.get(url, headers)
+    if response.status_code == 200:
+        data = response.json()
+        for result in data['results']:
+            output_list.append(result)
+
+        next_url = data['next']
+        if next_url is None:
+            return None
+        else:
+            fetch_departures(output_list, next_url, headers)
+    else:
+        print('[!] HTTP {0} calling [{1}]'.format(response.status_code, url))
+        return None
+    return output_list
+
+DEPARTURES = fetch_departures([], URL, HEADERS)
+
+
+####### Filter Data #######
+
+def filter_departures(departures, pivot_date, category):
+    """
+    Return departures with `start_date` later than `pivot_date`
+    and matching the given `category`.
+    """
+    output = []
+    for departure in departures:
+        start_date = datetime.strptime(departure['start_date'], '%Y-%m-%d')
+        cat = departure['category']
+        if start_date > pivot_date and cat == category:
+            output.append(departure)
+    return output
+
+FILTERED_DEPARTURES = filter_departures(DEPARTURES, datetime(2018, 6, 1), 'Adventurous')
+
+
+####### Write to CSV #######
+
+def write_departures_to_csv(departures):
+    """
+    Write departures to csv file with title-cased headers.
+    """
+    with open('filtered_departures.csv', 'w') as csvfile:
+        fieldnames = departures[0].keys()
+        writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
+
+        headers = [x.title().replace('_', ' ') for x in list(fieldnames)]
+        writer.writer.writerow(headers)
+        for departure in departures:
+            writer.writerow(departure)
+
+write_departures_to_csv(FILTERED_DEPARTURES)
diff --git a/filtered_departures.csv b/filtered_departures.csv
@@ -0,0 +1,25 @@
+Name,Start Date,Finish Date,Category
+New Zealand Encompassed,2018-08-31,2018-09-10,Adventurous
+Cambodia Overland,2018-08-31,2018-09-10,Adventurous
+Peru Overland,2018-08-31,2018-09-10,Adventurous
+Vietnam Safari,2018-08-31,2018-09-10,Adventurous
+Brazil Trek,2018-08-31,2018-09-10,Adventurous
+Cambodia Multisport,2018-08-31,2018-09-10,Adventurous
+Morocco Safari,2018-08-31,2018-09-10,Adventurous
+Galapagos Encompassed,2018-08-31,2018-09-10,Adventurous
+Galapagos Discovery,2018-08-31,2018-09-10,Adventurous
+New Zealand Adventure,2018-08-31,2018-09-10,Adventurous
+Galapagos Trek,2018-08-31,2018-09-10,Adventurous
+Brazil Encompassed,2018-08-31,2018-09-10,Adventurous
+Kenya Express,2018-08-31,2018-09-10,Adventurous
+Galapagos Discovery,2018-08-31,2018-09-10,Adventurous
+Sri Lanka Adventure,2018-08-31,2018-09-10,Adventurous
+Brazil Adventure,2018-08-31,2018-09-10,Adventurous
+Brazil Overland,2018-08-31,2018-09-10,Adventurous
+Kenya Encompassed,2018-08-31,2018-09-10,Adventurous
+Morocco Discovery,2018-08-31,2018-09-10,Adventurous
+Vietnam Trek,2018-08-31,2018-09-10,Adventurous
+New Zealand Safari,2018-08-31,2018-09-10,Adventurous
+Vietnam Encompassed,2018-08-31,2018-09-10,Adventurous
+Ethopia Express,2018-08-31,2018-09-10,Adventurous
+Sri Lanka Safari,2018-08-31,2018-09-10,Adventurous
