diff -r 94f586daa623 -r 8ca7f2cea729 alcatel/mediapartdb/MediapartReader.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/alcatel/mediapartdb/MediapartReader.py Thu Jan 24 16:58:55 2013 +0100 @@ -0,0 +1,120 @@ +''' +Created on 27 juin 2012 + +@author: gerard +''' +import MySQLdb +import logging +import warnings + +logger = logging.getLogger('document') + +class MediapartReader() : + def __init__(self): + self.conn = MySQLdb.connect (host = 'localhost', + user = 'mediapart', + passwd = 'mediapart', + db = 'www_mediapart_fr') + + warnings.filterwarnings('ignore', category=MySQLdb.Warning) + + def __del__(self): + self.conn.close () + + def get_date(self, document_id): + + self.conn.query('SELECT changed FROM node where nid=\''+document_id+'\'') + r = self.conn.use_result() + row = r.fetch_row() + if row: + date = (row)[0][0] + else: + date = '0' + + #changed_date = datetime.datetime.fromtimestamp(int(change_date_unix[0])).strftime('%Y-%m-%d %H:%M:%S') + return date + + def get_title(self, document_id): + self.conn.query('SELECT title FROM node where nid=\''+document_id+'\'') + r = self.conn.use_result() + row = r.fetch_row() + if row: + title = (row)[0][0].decode("windows-1252").encode("utf8") + else: + title = '' + return title + + '''def get_category(self, document_id): + self.conn.query('SELECT dst FROM url_alias where src="node/'+document_id+'"') + r = self.conn.use_result() + row = r.fetch_row() + if row: + dst = (row)[0][0] + fields = dst.split('/') + category = fields[0] + else: + category = '' + return category''' + + + def get_category(self, document_id): + cursor =self.conn.cursor() + cursor.execute('SELECT tid FROM term_node where nid=\''+document_id+'\'') + tids = cursor.fetchall() + category = '' + for tid in tids: + if str(tid[0]) == '27': + category = 'France' + return category + elif str(tid[0]) == '28': + category = 'International' + return category + elif str(tid[0]) == '29': + category = 'Economie' + return category + elif str(tid[0]) == '30': + category = 'Culture' + return category + else: + category = '' + return category + + def get_tags(self, document_id): + tags=[] + cursor = self.conn.cursor() + cursor.execute('SELECT tid FROM term_node where nid=\''+document_id+'\'') + tids = cursor.fetchall() + for tid in tids: + if str(tid[0]) != '27' and str(tid[0]) != '28' and str(tid[0]) != '29' and str(tid[0]) != '30': + cursor.execute('SELECT name FROM term_data where tid=' + str(tid[0])) + names = cursor.fetchall() + for name in names: + tags.append(str(name).decode("windows-1252").encode("utf8")) + return tags + + def get_url(self, document_id): + self.conn.query('SELECT dst FROM url_alias where src="node/'+document_id+'"') + r = self.conn.use_result() + row = r.fetch_row() + + if row: + print 'row!!!!!!!!!!!!!!!!!!!!!!!!' + print row + print (row)[0][0] + url = 'http://www.mediapart.fr/'+(row)[0][0] + else: + url = '' + return url + + def get_author(self, document_id): + self.conn.query('SELECT u.uid, u.name FROM users u, node n where u.uid=n.uid AND n.nid=\''+document_id+'\'') + r = self.conn.use_result() + row = r.fetch_row() + if row: + name = (row)[0][1].decode("windows-1252").encode("utf8") + uid = (row)[0][0] + else: + name = '' + uid='' + return {'id':uid, 'name':name} + \ No newline at end of file