Merge lp:~stefanor/ibid/faves-url into lp:~ibid-core/ibid/old-trunk-1.6

[Stefano Rivera]

[Stefano Rivera (stefanor)]

Stop gap solution

[lp:~stefanor/ibid/faves-url]

848. By Stefano Rivera on 2010-01-18

Merge from trunk

849. By Stefano Rivera on 2010-01-18

Fix Merge errors

850. By Stefano Rivera on 2010-01-18

Remove urlgrab_ prefix

851. By Stefano Rivera on 2010-01-18

Remove unused import

852. By Stefano Rivera on 2010-01-18

Whoops, debugging code left behind

853. By Stefano Rivera on 2010-01-24

Remove leading space

Unmerged revisions

=== modified file 'ibid.ini'
--- ibid.ini	2010-01-04 12:23:00 +0000
+++ ibid.ini	2010-01-18 21:57:14 +0000
@@ -77,9 +77,10 @@
         source = atrum
         channel = "#ibid"
         server = localhost
-    [[url]]
-        delicious_username = ibidtest
-        delicious_password = a123456
+    [[urlgrab]]
+        urlgrab_username = ibidtest
+        urlgrab_password = a123456
+        urlgrab_service = delicious
 
 [databases]
     ibid = sqlite:///ibid.db

=== modified file 'ibid/plugins/urlgrab.py'
--- ibid/plugins/urlgrab.py	2010-01-16 18:13:23 +0000
+++ ibid/plugins/urlgrab.py	2010-01-18 21:57:14 +0000
@@ -1,8 +1,7 @@
 from datetime import datetime
 from httplib import BadStatusLine
 from urllib import urlencode
-from urllib2 import urlopen, build_opener, HTTPError, HTTPBasicAuthHandler, \
-        install_opener
+from urllib2 import urlopen, build_opener, HTTPError, HTTPBasicAuthHandler
 import logging
 import re
 
@@ -19,7 +18,8 @@
 
 log = logging.getLogger('plugins.urlgrab')
 
-help['url'] = u'Captures URLs seen in channel to database and/or to delicious, and shortens and lengthens URLs'
+help['url'] = u'Captures URLs seen in channel to database and/or to ' \
+              u'delicious/faves'
 
 class URL(Base):
     __table__ = Table('urls', Base.metadata,
@@ -48,8 +48,49 @@
         self.identity_id = identity_id
         self.time = datetime.utcnow()
 
-class Delicious(object):
-    def add_post(self, username, password, event, url=None):
+class Grab(Processor):
+    addressed = False
+    processed = True
+
+    username  = Option('urlgrab_username', 'Account name for URL posting')
+    password  = Option('urlgrab_password', 'Password for URL Posting')
+    service   = Option('urlgrab_service',
+                       'URL Posting Service (delicious/faves)', 'delicious')
+
+    def setup(self):
+        if resource_exists(__name__, '../data/tlds-alpha-by-domain.txt'):
+            tlds = [tld.strip().lower() for tld
+                    in resource_stream(__name__, '../data/tlds-alpha-by-domain.txt')
+                        .readlines()
+                    if not tld.startswith('#')
+            ]
+
+        else:
+            log.warning(u"Couldn't open TLD list, falling back to minimal default")
+            tlds = 'com.org.net.za'.split('.')
+
+        self.grab.im_func.pattern = re.compile((
+            r'(?:[^@./]\b(?!\.)|\A)('       # Match a boundary, but not on an e-mail address
+            r'(?:\w+://|(?:www|ftp)\.)\S+?' # Match an explicit URL or guess by www.
+            r'|[^@\s:/]+\.(?:%s)(?:/\S*?)?' # Guess at the URL based on TLD
+            r')[\[>)\]"\'.,;:]*(?:\s|\Z)'   # End boundary
+        ) % '|'.join(tlds), re.I | re.DOTALL)
+
+    @handler
+    def grab(self, event, url):
+        if url.find('://') == -1:
+            if url.lower().startswith('ftp'):
+                url = 'ftp://%s' % url
+            else:
+                url = 'http://%s' % url
+
+        u = URL(url, event.channel, event.identity)
+        event.session.save_or_update(u)
+
+        if self.service and self.username:
+            self._post_url(event, url)
+
+    def _post_url(self, event, url=None):
         "Posts a URL to delicious.com"
 
         date = datetime.utcnow()
@@ -75,7 +116,8 @@
             obfusc_conn = at_re.sub('^', connection_body[1])
             obfusc_chan = at_re.sub('^', event.channel)
 
-        tags = u' '.join((event.sender['nick'], obfusc_conn, obfusc_chan, event.source))
+        tags = u' '.join((event.sender['nick'], obfusc_conn, obfusc_chan,
+                          event.source))
 
         data = {
             'url' : url.encode('utf-8'),
@@ -86,18 +128,39 @@
             'extended' : event.message['raw'].encode('utf-8'),
             }
 
-        self._set_auth(username, password)
-        posturl = 'https://api.del.icio.us/v1/posts/add?' + urlencode(data)
+        if self.service.lower() == 'delicious':
+            service = ('del.icio.us API', 'https://api.del.icio.us')
+        elif self.service.lower() == 'faves':
+            service = ('Faves', 'https://secure.faves.com')
+        else:
+            log.error(u'Unknown social bookmarking service: %s', self.service)
+            return
+        auth_handler = HTTPBasicAuthHandler()
+        auth_handler.add_password(service[0], service[1],
+                                  self.username, self.password)
+        opener = build_opener(auth_handler)
+
+        posturl = service[1] + '/v1/posts/add?' + urlencode(data)
 
         try:
-            resp = urlopen(posturl).read()
+            resp = opener.open(posturl).read()
             if 'done' in resp:
-                log.debug(u"Posted url '%s' to delicious, posted in %s on %s by %s/%i (%s)",
-                         url, event.channel, event.source, event.account, event.identity, event.sender['connection'])
+                log.debug(u"Posted url '%s' to %s, posted in %s on %s "
+                          u"by %s/%i (%s)",
+                          url, self.service, event.channel, event.source,
+                          event.account, event.identity,
+                          event.sender['connection'])
             else:
-                log.error(u"Error posting url '%s' to delicious: %s", url, resp)
+                log.error(u"Error posting url '%s' to %s: %s",
+                          url, self.service, resp)
+        except HTTPError, e:
+            if e.code == 401:
+                log.error(u"Incorrect password for %s, couldn't post",
+                          self.service)
+            print e.hdrs
         except BadStatusLine, e:
-            log.error(u"Error posting url '%s' to delicious: %s", url, unicode(e))
+            log.error(u"Error posting url '%s' to %s: %s",
+                      url, self.service, unicode(e))
 
     def _get_title(self, url):
         "Gets the title of a page"
@@ -112,53 +175,4 @@
             log.debug(u"Error determining title for %s: %s", url, unicode(e))
             return url
 
-    def _set_auth(self, username, password):
-        "Provides HTTP authentication on username and password"
-        auth_handler = HTTPBasicAuthHandler()
-        auth_handler.add_password('del.icio.us API', 'https://api.del.icio.us', username, password)
-        opener = build_opener(auth_handler)
-        install_opener(opener)
-
-class Grab(Processor):
-    addressed = False
-    processed = True
-
-    username  = Option('delicious_username', 'delicious account name')
-    password  = Option('delicious_password', 'delicious account password')
-    delicious = Delicious()
-
-    def setup(self):
-        if resource_exists(__name__, '../data/tlds-alpha-by-domain.txt'):
-            tlds = [tld.strip().lower() for tld
-                    in resource_stream(__name__, '../data/tlds-alpha-by-domain.txt')
-                        .readlines()
-                    if not tld.startswith('#')
-            ]
-
-        else:
-            log.warning(u"Couldn't open TLD list, falling back to minimal default")
-            tlds = 'com.org.net.za'.split('.')
-
-        self.grab.im_func.pattern = re.compile((
-            r'(?:[^@./]\b(?!\.)|\A)('       # Match a boundary, but not on an e-mail address
-            r'(?:\w+://|(?:www|ftp)\.)\S+?' # Match an explicit URL or guess by www.
-            r'|[^@\s:/]+\.(?:%s)(?:/\S*?)?' # Guess at the URL based on TLD
-            r')[\[>)\]"\'.,;:]*(?:\s|\Z)'   # End boundary
-        ) % '|'.join(tlds), re.I | re.DOTALL)
-
-    @handler
-    def grab(self, event, url):
-        if url.find('://') == -1:
-            if url.lower().startswith('ftp'):
-                url = 'ftp://%s' % url
-            else:
-                url = 'http://%s' % url
-
-        u = URL(url, event.channel, event.identity)
-        event.session.save_or_update(u)
-
-        if self.username != None:
-            self.delicious.add_post(self.username, self.password, event, url)
-
-
-# vi: set et sta sw=4 ts=4:
+ # vi: set et sta sw=4 ts=4: