home *** CD-ROM | disk | FTP | other *** search
- # -*- Mode: python; coding: utf-8; tab-width: 8; indent-tabs-mode: t; -*-
- #
- # Copyright (C) 2006 - Jonathan Matthew
- #
- # This program is free software; you can redistribute it and/or modify
- # it under the terms of the GNU General Public License as published by
- # the Free Software Foundation; either version 2, or (at your option)
- # any later version.
- #
- # This program is distributed in the hope that it will be useful,
- # but WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- # GNU General Public License for more details.
- #
- # You should have received a copy of the GNU General Public License
- # along with this program; if not, write to the Free Software
- # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
-
- import gobject
-
- try:
- import gnomevfs
- use_gnomevfs = True
- except:
- import urllib
- use_gnomevfs = False
-
-
- class GnomeVFSAsyncSrc (object):
- def __init__ (self):
- self.chunk = 4096
-
- def read_cb (self, handle, buffer, exc_type, bytes_requested, (data, callback, args)):
- if exc_type:
- if issubclass (exc_type, gnomevfs.EOFError):
- gobject.idle_add (callback, data, *args)
- handle.close (lambda *args: None)
- else:
- gobject.idle_add (callback, None, *args)
- handle.close (lambda *args: None)
- return
-
- data += buffer
- handle.read (self.chunk, self.read_cb, (data, callback, args))
-
- def open_cb (self, handle, exc_type, (data, callback, args)):
- if exc_type:
- gobject.idle_add (callback, None, *args)
- return
-
- handle.read (self.chunk, self.read_cb, (data, callback, args))
-
- def get_url (self, url, callback, *args):
- gnomevfs.async.open (url, self.open_cb, data=("", callback, args))
-
-
- class URLLibSrc (object):
- def get_url (self, url, callback, *args):
- try:
- sock = urllib.urlopen (url)
- data = sock.read ()
- sock.close ()
- callback (data, *args)
- except:
- callback (None, *args)
- raise
-
-
- def Loader ():
- if use_gnomevfs:
- return GnomeVFSAsyncSrc ()
- else:
- return URLLibSrc ()
-