Progress toward sample implementation.

tseaver · Aug 28, 2014 · Oct 3, 2014 · Oct 3, 2014 · Oct 9, 2014 · Oct 9, 2014
commit 943f9df93b47627823968fcf55ca3d296504fca8
diff --git a/docs/datastore-narrative_example.rst b/docs/datastore-narrative_example.rst
@@ -48,7 +48,7 @@ Sally can review a submitted expense report using the its ID:
 
 .. code-block:: bash
 
-   $ review_expenses show --report-id=sally/expenses-2014-09-01
+   $ review_expenses show --employee-id=sally --report-idexpenses-2014-09-01
    Report-ID: sally/expenses-2014-09-01
    Report-Status: Pending
    Employee-ID: sally
@@ -63,7 +63,7 @@ While in "Pending" status, Sally can edit the CSV and resubmit it:
 
 .. code-block:: bash
 
-   $ submit_expenses update --report-id=sally/expenses-2014-09-01 expenses-20140901-amended.csv
+   $ submit_expenses update --employee-id=sally --report-idexpenses-2014-09-01 expenses-20140901-amended.csv
    Processed 15 rows.
    Updated, report ID: sally/expenses-2014-09-01
 
@@ -73,20 +73,20 @@ Sally's boss, Pat, can review all open expense reports:
 
    $ review_expenses list --status=Pending
 
-   "Report ID","Description","Status","Memo"
-   "sally/2014-09-01","Frotz project kickoff, San Jose","Pending",""
+   "Employee ID","Report ID","Description","Status","Memo"
+   "sally","2014-09-01","Frotz project kickoff, San Jose","Pending",""
 
 and download the expenses just as above.  Pat can approve Sally's expense
 report:
 
 .. code-block:: bash
 
-   $ review_expenses approve --report-id=sally/expenses-2014-09-01 --check-number=4093
+   $ review_expenses approve --employee-id=sally --report-idexpenses-2014-09-01 --check-number=4093
    Approved, report ID: sally/expenses-2014-09-01, check #4093
 
 or reject it:
 
 .. code-block:: bash
 
-   $ review_expenses reject --report-id=sally/expenses-2014-09-01 --reason="Travel not authorized by client"
+   $ review_expenses reject --employee-id=sally --report-idexpenses-2014-09-01 --reason="Travel not authorized by client"
    Rejected, report ID: sally/expenses-2014-09-01, reason: Travel not authorized by client
diff --git a/docs/examples/expenses/expenses/__init__.py b/docs/examples/expenses/expenses/__init__.py
@@ -1 +1,50 @@
 # package
+import os
+
+from gcloud import dataset
+from gcloud.dataset.key import Key
+from gcloud.dataset.query import Query
+
+def get_dataset():
+    client_email = os.environ['EXPENSES_CLIENT_EMAIL']
+    private_key_path = os.environ['EXPENSES_PRIVATE_KEY_PATH']
+    dataset_id = os.environ['EXPENSES_DATASET_ID']
+    conn = dataset.get_connection(client_email, private_key_path)
+    return conn.dataset_id(dataset_id)
+
+def get_employee(dataset, employee_id):
+    key = Key(dataset, path=[{'kind': 'Employee', 'name': employee_id}])
+    employee = dataset.get_entity(key)
+    if employee is None:
+        employee = dataset.entity('Employee').key(key)
+        employee.save()
+    return employee
+
+def get_report(dataset, employee_id, report_id):
+    key = Key(dataset,
+              path=[{'kind': 'Employee', 'name': employee_id},
+                    {'kind': 'Expense Report', 'name': report_id},
+                   ])
+    report = dataset.get_entity(key)
+    if report is None:
+        report = dataset.entity('Report').key(key)
+        report.save()
+    return report
+
+def upsert_report(employee_id, report_id, rows):
+    dataset = get_dataset()
+    with dataset.transaction():
+        employee = get_employee(dataset, employee_id)
+        report = get_report(dataset, employee_id, report_id)
+        query = Query('Expense Item', dataset.connection())
+        # Delete any existing items.
+        for existing in query.ancestor(report.key()).fetch():
+            existing.delete()
+        # Add items based on rows.
+        for i, row in enumerate(rows):
+            path = report.key.path() + [{'kind': 'Expense Item', 'id': i}]
+            key = Key(dataset, path=path)
+            item = dataset.entity('Expense Item').key(key)
+            for k, v in rows.items():
+                item[k] = v
+            item.save()