removed support for contact import from google (@moyamo, did you added this again?)

This commit is contained in:
Steffen Vogel 2015-12-03 13:48:25 +01:00
parent a84e7258d5
commit f5a668e589

81
bot.py
View file

@ -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