From f0e5b243c657f56488b74edde270e2ac84ade24f Mon Sep 17 00:00:00 2001
From: Andrew Melton <andrew.melton@rackspace.com>
Date: Wed, 5 Mar 2014 14:34:34 -0500
Subject: [PATCH] Using partial instead of defining a new func

---
 reports/nova_usage_audit.py | 7 ++++---
 reports/usage_audit.py      | 2 +-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/reports/nova_usage_audit.py b/reports/nova_usage_audit.py
index ca11475..ae2bd2b 100644
--- a/reports/nova_usage_audit.py
+++ b/reports/nova_usage_audit.py
@@ -20,6 +20,7 @@
 
 import argparse
 import datetime
+import functools
 import json
 import sys
 import os
@@ -197,9 +198,9 @@ def audit_for_period(beginning, ending, ums=False, ums_offset=0):
     ending_decimal = dt.dt_to_decimal(ending)
 
     if ums:
-        def verifier_audit_func(start, end, model):
-            return usage_audit._verifier_audit_for_day_ums(start, end, model,
-                                                           ums_offset)
+        verifier_audit_func = functools.partial(
+            usage_audit._verifier_audit_for_day_ums, ums_offset=ums_offset
+        )
     else:
         verifier_audit_func = usage_audit._verifier_audit_for_day
 
diff --git a/reports/usage_audit.py b/reports/usage_audit.py
index 4eed669..965d2ac 100644
--- a/reports/usage_audit.py
+++ b/reports/usage_audit.py
@@ -94,7 +94,7 @@ def _verifier_audit_for_day(beginning, ending, exists_model):
     return _verified_audit_base(base_query, exists_model)
 
 
-def _verifier_audit_for_day_ums(beginning, ending, exists_model, ums_offset):
+def _verifier_audit_for_day_ums(beginning, ending, exists_model, ums_offset=0):
     # NOTE(apmelton):
     # This is the UMS query we're trying to match.
     # where (