removed support for contact import from google (@moyamo, did you added this again?)
This commit is contained in:
parent
a84e7258d5
commit
f5a668e589
81
bot.py
81
bot.py
|
@ -31,19 +31,12 @@ import os
|
||||||
import utils
|
import utils
|
||||||
|
|
||||||
from constants import *
|
from constants import *
|
||||||
#from googleclient import GoogleClient
|
|
||||||
|
|
||||||
#from Yowsup.Contacts.contacts import WAContactsSyncRequest
|
|
||||||
|
|
||||||
class Bot():
|
class Bot():
|
||||||
def __init__(self, session, name = "Bot"):
|
def __init__(self, session, name = "Bot"):
|
||||||
self.session = session
|
self.session = session
|
||||||
self.name = name
|
self.name = name
|
||||||
|
|
||||||
#self.google = GoogleClient()
|
|
||||||
|
|
||||||
self.commands = {
|
self.commands = {
|
||||||
"import": self._import,
|
|
||||||
"help": self._help,
|
"help": self._help,
|
||||||
"prune": self._prune,
|
"prune": self._prune,
|
||||||
"welcome": self._welcome,
|
"welcome": self._welcome,
|
||||||
|
@ -81,81 +74,7 @@ class Bot():
|
||||||
def send(self, message):
|
def send(self, message):
|
||||||
self.session.backend.handleMessage(self.session.user, self.name, message)
|
self.session.backend.handleMessage(self.session.user, self.name, message)
|
||||||
|
|
||||||
def __do_import(self, token):
|
|
||||||
# Google
|
|
||||||
google = self.google.getContacts(token)
|
|
||||||
self.send("%d buddies imported from google" % len(google))
|
|
||||||
|
|
||||||
result = { }
|
|
||||||
for number, name in google.iteritems():
|
|
||||||
number = re.sub("[^0-9]", "", number)
|
|
||||||
number = number if number[0] == "0" else "+" + number
|
|
||||||
|
|
||||||
result[number] = { 'nick': name, 'state': 0 }
|
|
||||||
|
|
||||||
# WhatsApp
|
|
||||||
user = self.session.legacyName
|
|
||||||
password = self.session.password
|
|
||||||
sync = WAContactsSyncRequest(user, password, result.keys())
|
|
||||||
whatsapp = sync.send()['c']
|
|
||||||
|
|
||||||
for w in whatsapp:
|
|
||||||
result[w['p']]['state'] = w['w']
|
|
||||||
result[w['p']]['number'] = w['n']
|
|
||||||
|
|
||||||
self.send("%d buddies are using whatsapp" % len(filter(lambda w: w['w'], whatsapp)))
|
|
||||||
|
|
||||||
for r in result.values():
|
|
||||||
if r['nick']:
|
|
||||||
self.session.buddies.add(
|
|
||||||
number = r['number'],
|
|
||||||
nick = r['nick'],
|
|
||||||
groups = [u'Google'],
|
|
||||||
state = r['state']
|
|
||||||
)
|
|
||||||
|
|
||||||
self.send("%d buddies imported" % len(whatsapp))
|
|
||||||
|
|
||||||
def __get_token(self, filename, timeout = 30):
|
|
||||||
file = open(filename, 'r')
|
|
||||||
file.seek(-1, 2) # look at the end
|
|
||||||
|
|
||||||
count = 0
|
|
||||||
while count < timeout:
|
|
||||||
line = file.readline()
|
|
||||||
|
|
||||||
if line in ["", "\n"]:
|
|
||||||
time.sleep(1)
|
|
||||||
count += 1
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
timestamp, number, token = line[:-1].split("\t")
|
|
||||||
if (number == self.session.legacyName):
|
|
||||||
file.close()
|
|
||||||
return token
|
|
||||||
|
|
||||||
file.close()
|
|
||||||
|
|
||||||
# commands
|
# commands
|
||||||
def _import(self, token = None):
|
|
||||||
if not token:
|
|
||||||
token_url = self.google.getTokenUrl("http://whatsapp.0l.de/auth.py")
|
|
||||||
auth_url = "http://whatsapp.0l.de/auth.py?number=%s&auth_url=%s" % (self.session.legacyName, urllib.quote(token_url))
|
|
||||||
short_url = utils.shorten(auth_url)
|
|
||||||
self.send("please visit this url to auth: %s" % short_url)
|
|
||||||
|
|
||||||
self.send("waiting for authorization...")
|
|
||||||
token = self.__get_token(TOKEN_FILE)
|
|
||||||
if token:
|
|
||||||
self.send("got token: %s" % token)
|
|
||||||
self.__do_import(token)
|
|
||||||
self.session.updateRoster()
|
|
||||||
else:
|
|
||||||
self.send("timeout! please use \"\\import [token]\"")
|
|
||||||
else:
|
|
||||||
self.__do_import(token)
|
|
||||||
self.session.updateRoster()
|
|
||||||
|
|
||||||
def _sync(self):
|
def _sync(self):
|
||||||
user = self.session.legacyName
|
user = self.session.legacyName
|
||||||
password = self.session.password
|
password = self.session.password
|
||||||
|
|
Loading…
Reference in a new issue