-

cool-RR · cool-RR · commit 3b6537e8cd01 · 2013-08-25T23:07:48.000+03:00
diff --git a/python_toolbox/cute_profile/cute_profile.py b/python_toolbox/cute_profile/cute_profile.py
@@ -14,6 +14,7 @@
 from python_toolbox import decorator_tools
 
 from . import base_profile
+from . import profile_handling
 
 
 def profile(statement, globals_, locals_):
@@ -98,7 +99,11 @@ def inner(function_, *args, **kwargs):
                     globals(), locals()
                 )
 
-                if misc_tools.is_legal_email_address(profile_handler)
+                
+                profile_handler = \
+                          profile_handling.get_profile_handler(profile_handler)
+                
+                profile_handler(profile_result)
                 
                 Z Z Z Do shit depending on `profile_handler`. Allow filename,  folder,  email address, or index for  printing (sort). In any case do everything on a thread.
                 
diff --git a/python_toolbox/cute_profile/profile_handling.py b/python_toolbox/cute_profile/profile_handling.py
@@ -2,16 +2,28 @@
 # This program is distributed under the MIT license.
 
 import threading
+import datetime as datetime_module
+import os.path
+import pstats
 
 import envelopes
 
+from python_toolbox import caching
+from python_toolbox import misc_tools
+
+from . import base_profile
+
 
 class BaseProfileHandler(object):
     __metaclass__ = abc.ABCMeta
     
     @abc.abstractmethod
     def __call__(self):
         pass
+    
+    default_file_name = caching.CachedProperty(
+        lambda self: 'profile-%s' % datetime_module.datetime.now()
+    )
         
     
 
@@ -28,33 +40,68 @@ def thread_job(self):
         pass
     
 
-class EmailProfileHandler(BaseProfileHandler):
+class EmailProfileHandler(AuxiliaryThreadProfileHandler):
     def __init__(self, email_address, smtp_server, smtp_user, smtp_password,
                  use_tls=True):
         
+        if use_tls == 'False':
+            use_tls = False
+        
         self.email_address = email_address
+        self.smtp_server = smtp_server
+        self.smtp_user = smtp_user
+        self.smtp_password = smtp_password
+        self.use_tls = use_tls
         
-        s
+    def thread_job(self, profile_data):
+        envelope = envelopes.Envelope(
+            to_addr=self.email_address,
+            subject='Profile data', 
+        )
         
-    def thread_job(self):
-        envelope = envelopes.Envelope(to_addr=self.email_address,
-                                      subject='Profile data')
+        envelope.add_attachment_from_memory(profile_data,
+                                            self.default_file_name, 
+                                            'application/octet-stream')
         
-        envelope.add_attachment('/Users/bilbo/Pictures/helicopter.jpg')
+        envelope.send(self.smtp_server, login=self.smtp_user,
+                      password=self.smtp_password, tls=self.use_tls)
         
-        envelope.send('smtp.googlemail.com', login='from@example.com',
-                      password='password', tls=True)
         
-        ('profile-%s-%s' % (cleaned_path,
-                            datetime_tools.get_now()),
-         profile_data, 
-         'application/octet-stream'), 
 
 
+class FolderProfileHandler(AuxiliaryThreadProfileHandler):
+    
+    def __init__(self, folder_path):
+        self.folder_path = folder_path
+        
+    def thread_job(self, profile_data):
+        with open(os.path.join(self.folder_path, self.default_file_name), \
+                                                          'wb') as output_file:
+            output_file.write(profile_data)
+        
+
 
+class PrintProfileHandler(BaseProfileHandler):
+    
+    def __init__(self, sort_order):
+        self.sort_order = sort_order
+        
+    def __call__(self, profile_data):
+        pstats.Stats(self).strip_dirs().sort_stats(self.sort_order). \
+                                                                  print_stats()
+        
 
 
 def get_profile_handler(profile_handler_string):
-    
-    
-    
+    if misc_tools.is_legal_email_address(profile_handler_string.split('\n')
+                                                                          [0]):
+        return EmailProfileHandler(*profile_handler_string.split('\n'))
+    elif os.path.isdir(profile_handler_string):
+        return FolderProfileHandler(profile_handler_string)
+    else:
+        assert profile_handler_string == '' or int(profile_handler_string)
+        try:
+            sort_order = int(profile_handler_string)
+        except ValueError:
+            sort_order = -1
+        return PrintProfileHandler(sort_order)
