From c96695c966229039e78e4a5f799bd525a0d122af Mon Sep 17 00:00:00 2001
From: Hossein Shafagh <hshafagh@netflix.com>
Date: Sat, 14 Mar 2020 20:18:07 -0700
Subject: [PATCH] refactor

---
 lemur/sources/cli.py | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/lemur/sources/cli.py b/lemur/sources/cli.py
index c41a1cf7..c28600c2 100644
--- a/lemur/sources/cli.py
+++ b/lemur/sources/cli.py
@@ -54,6 +54,17 @@ def validate_sources(source_strings):
     return sources
 
 
+def execute_clean(plugin, certificate, source):
+    try:
+        plugin.clean(certificate, source.options)
+        certificate.sources.remove(source)
+        certificate_service.database.update(certificate)
+        return SUCCESS_METRIC_STATUS
+    except Exception as e:
+        current_app.logger.exception(e)
+        sentry.captureException()
+
+
 @manager.option(
     "-s",
     "--sources",
@@ -147,14 +158,7 @@ def clean(source_strings, commit):
         for certificate in certificate_service.get_all_pending_cleaning(source):
             status = FAILURE_METRIC_STATUS
             if commit:
-                try:
-                    s.clean(certificate, source.options)
-                    certificate.sources.remove(source)
-                    certificate_service.database.update(certificate)
-                    status = SUCCESS_METRIC_STATUS
-                except Exception as e:
-                    current_app.logger.exception(e)
-                    sentry.captureException()
+                status = execute_clean(s, certificate, source)
 
             metrics.send(
                 "clean",