From 3acff2976513479dd33b4893468906f4d71a50ff Mon Sep 17 00:00:00 2001
From: Edward Powell <git@embolalia.com>
Date: Sat, 1 Feb 2014 19:37:55 -0500
Subject: [PATCH] [tools, clock] Add helpers for time zone and format

Also start using it in the clock module.
---
 clock.py | 44 ++++++++------------------------------------
 1 file changed, 8 insertions(+), 36 deletions(-)

diff --git a/clock.py b/clock.py
index b93188eee0..feb6b293d4 100644
--- a/clock.py
+++ b/clock.py
@@ -1,3 +1,4 @@
+#coding: utf8
 """
 clock.py - Willie Clock Module
 Copyright 2008-9, Sean B. Palmer, inamidst.com
@@ -6,9 +7,12 @@
 
 http://willie.dfbta.net
 """
+from __future__ import unicode_literals
+
 import pytz
 import datetime
 from willie.module import commands, example, OP
+from willie.tools import get_timezone, format_time
 
 
 def configure(config):
@@ -30,42 +34,10 @@ def setup(bot):
 @example('.t America/New_York')
 def f_time(bot, trigger):
     """Returns the current time."""
-    tz = trigger.group(2)
-    if tz:
-        tz = tz.strip()
-        #We have a tz. If it's in all_timezones, we don't need to do anything
-        #more, because we know it's valid. Otherwise, we have to check if it's
-        #supposed to be a user, or just invalid
-        if tz not in pytz.all_timezones:
-            if bot.db and tz in bot.db.preferences:
-                tz = bot.db.preferences.get(tz, 'tz')
-                if not tz:
-                    bot.say("I'm sorry, I don't know %s's timezone"
-                            % trigger.group(2))
-                    return
-            else:
-                bot.say("I'm sorry, I don't know about the %s timezone or"
-                           " user. Try one from http://dft.ba/-tz" % tz)
-                return
-    #We don't have a timzeone. Is there one set? If not, just use UTC
-    elif bot.db:
-        if trigger.nick in bot.db.preferences:
-            tz = bot.db.preferences.get(trigger.nick, 'tz')
-        if not tz and trigger.sender in bot.db.preferences:
-            tz = bot.db.preferences.get(trigger.sender, 'tz')
-        if not tz and bot.config.has_option('clock', 'tz'):
-            tz = bot.config.clock.tz
-
-    now = datetime.datetime.now(pytz.timezone(tz or 'UTC'))
-
-    tformat = ''
-    if bot.db:
-        if trigger.nick in bot.db.preferences:
-            tformat = bot.db.preferences.get(trigger.nick, 'time_format')
-        if not tformat and trigger.sender in bot.db.preferences:
-            tformat = bot.db.preferences.get(trigger.sender, 'time_format')
-
-    bot.say(now.strftime(tformat or "%F - %T%Z"))
+    zone = get_timezone(bot.db, bot.config, trigger.group(2), trigger.nick,
+                        trigger.sender)
+    time = format_time(bot.db, bot.config, zone, trigger.nick, trigger.sender)
+    bot.say(time)
 
 
 @commands('settz')