204 lines
7.5 KiB
Python
204 lines
7.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2014 Roger Philibert
|
|
#
|
|
# This file is part of weboob.
|
|
#
|
|
# weboob is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# weboob 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 Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
import datetime
|
|
|
|
from weboob.capabilities.messages import CapMessages, CapMessagesPost, Thread, Message
|
|
from weboob.capabilities.dating import CapDating, Optimization
|
|
from weboob.capabilities.account import CapAccount, StatusField
|
|
from weboob.tools.backend import Module, BackendConfig
|
|
from weboob.tools.value import Value, ValueBackendPassword
|
|
from weboob.tools.date import local2utc
|
|
from weboob.tools.log import getLogger
|
|
|
|
from .browser import PlayMeBrowser, FacebookBrowser, NoCredits
|
|
|
|
|
|
__all__ = ['PlayMeModule']
|
|
|
|
|
|
class ProfilesWalker(Optimization):
|
|
def __init__(self, sched, storage, browser):
|
|
self._sched = sched
|
|
self._storage = storage
|
|
self._browser = browser
|
|
self._logger = getLogger('walker', browser.logger)
|
|
|
|
self._view_cron = None
|
|
|
|
def start(self):
|
|
self._view_cron = self._sched.schedule(1, self.view_profile)
|
|
return True
|
|
|
|
def stop(self):
|
|
self._sched.cancel(self._view_cron)
|
|
self._view_cron = None
|
|
return True
|
|
|
|
def set_config(self, params):
|
|
pass
|
|
|
|
def is_running(self):
|
|
return self._view_cron is not None
|
|
|
|
def view_profile(self):
|
|
delay = 900
|
|
try:
|
|
challenged = self._storage.get('challenged', default=[])
|
|
for user in self._browser.find_users(48.883989, 2.367168):
|
|
if user['id'] in challenged:
|
|
continue
|
|
|
|
try:
|
|
self._browser.challenge(user['id'])
|
|
except NoCredits as e:
|
|
delay = int(str(e))
|
|
self._logger.info('No more credits (next try in %d minutes)' % (delay/60))
|
|
else:
|
|
self._logger.info('Challenged %s' % user['name'])
|
|
challenged.append(user['id'])
|
|
self._storage.set('challenged', challenged)
|
|
self._storage.save()
|
|
break
|
|
finally:
|
|
if self._view_cron is not None:
|
|
self._view_cron = self._sched.schedule(delay, self.view_profile)
|
|
|
|
|
|
class PlayMeModule(Module, CapMessages, CapMessagesPost, CapDating, CapAccount):
|
|
NAME = 'playme'
|
|
DESCRIPTION = u'PlayMe dating mobile application'
|
|
MAINTAINER = u'Roger Philibert'
|
|
EMAIL = 'roger.philibert@gmail.com'
|
|
LICENSE = 'AGPLv3+'
|
|
VERSION = '1.1'
|
|
CONFIG = BackendConfig(Value('username', label='Facebook email'),
|
|
ValueBackendPassword('password', label='Facebook password'))
|
|
|
|
BROWSER = PlayMeBrowser
|
|
STORAGE = {'contacts': {},
|
|
'challenged': [],
|
|
}
|
|
|
|
def create_default_browser(self):
|
|
facebook = FacebookBrowser()
|
|
facebook.login(self.config['username'].get(),
|
|
self.config['password'].get())
|
|
return PlayMeBrowser(facebook)
|
|
|
|
# ---- CapDating methods -----------------------
|
|
|
|
def init_optimizations(self):
|
|
self.add_optimization('PROFILE_WALKER', ProfilesWalker(self.weboob.scheduler, self.storage, self.browser))
|
|
|
|
# ---- CapMessages methods ---------------------
|
|
|
|
def fill_thread(self, thread, fields):
|
|
return self.get_thread(thread)
|
|
|
|
def iter_threads(self):
|
|
for thread in self.browser.get_threads():
|
|
t = Thread(thread['id'])
|
|
t.flags = Thread.IS_DISCUSSION
|
|
t.title = u'Discussion with %s' % thread['name']
|
|
t.date = local2utc(datetime.datetime.fromtimestamp(thread['last_message']['utc_timestamp']))
|
|
yield t
|
|
|
|
def get_thread(self, thread):
|
|
if not isinstance(thread, Thread):
|
|
thread = Thread(thread)
|
|
thread.flags = Thread.IS_DISCUSSION
|
|
|
|
user = self.browser.get_user(thread.id)
|
|
thread.title = u'Discussion with %s' % user['name']
|
|
|
|
contact = self.storage.get('contacts', thread.id, default={'lastmsg': 0})
|
|
|
|
signature = u'Age: %s' % user['age']
|
|
signature += u'\nLast online: %s' % user['last_online']
|
|
signature += u'\nPhotos:\n\t%s' % '\n\t'.join([user['photo_host'] + photo['large'] for photo in user['photos']])
|
|
|
|
child = None
|
|
|
|
for msg in self.browser.get_thread_messages(thread.id):
|
|
flags = 0
|
|
if int(contact['lastmsg']) < msg['utc_timestamp']:
|
|
flags = Message.IS_UNREAD
|
|
|
|
if msg['type'] == 'msg':
|
|
content = unicode(msg['msg'])
|
|
elif msg['type'] == 'new_challenge':
|
|
content = u'A new challenge has been proposed!'
|
|
elif msg['type'] == 'serie':
|
|
content = u"I've played"
|
|
elif msg['type'] == 'end_game':
|
|
content = u'%s is the winner! (%s VS %s)' % (self.browser.my_name if msg['score']['w'] == self.browser.my_id else user['name'], msg['score']['s'][0], msg['score']['s'][1])
|
|
else:
|
|
content = u'Unknown action: %s' % msg['type']
|
|
|
|
msg = Message(thread=thread,
|
|
id=msg['utc_timestamp'],
|
|
title=thread.title,
|
|
sender=unicode(self.browser.my_name if msg['from'] == self.browser.my_id else user['name']),
|
|
receivers=[unicode(self.browser.my_name if msg['from'] != self.browser.my_id else user['name'])],
|
|
date=local2utc(datetime.datetime.fromtimestamp(msg['utc_timestamp'])),
|
|
content=content,
|
|
children=[],
|
|
parent=None,
|
|
signature=signature if msg['from'] != self.browser.my_id else u'',
|
|
flags=flags)
|
|
|
|
if child:
|
|
msg.children.append(child)
|
|
child.parent = msg
|
|
child = msg
|
|
thread.root = child
|
|
|
|
return thread
|
|
|
|
def iter_unread_messages(self):
|
|
for thread in self.iter_threads():
|
|
thread = self.get_thread(thread)
|
|
for message in thread.iter_all_messages():
|
|
if message.flags & message.IS_UNREAD:
|
|
yield message
|
|
|
|
def set_message_read(self, message):
|
|
contact = self.storage.get('contacts', message.thread.id, default={'lastmsg': 0})
|
|
if int(contact['lastmsg']) < int(message.id):
|
|
contact['lastmsg'] = int(message.id)
|
|
self.storage.set('contacts', message.thread.id, contact)
|
|
self.storage.save()
|
|
|
|
# ---- CapMessagesPost methods ---------------------
|
|
|
|
def post_message(self, message):
|
|
self.browser.post_message(message.thread.id, message.content)
|
|
|
|
# ---- CapAccount methods ---------------------
|
|
|
|
def get_account_status(self):
|
|
return (
|
|
StatusField(u'myname', u'My name', unicode(self.browser.my_name)),
|
|
StatusField(u'credits', u'Credits', unicode(self.browser.credits)),
|
|
)
|
|
|
|
OBJECTS = {Thread: fill_thread,
|
|
}
|