0

this is my first question here. I try to rewrite telegram bot in Python. And I cant solve one problem. When I test my bot by myself everything fine, but if another user connect with my bot script doesn`t start for him with initial state of variables. For example, in this code script increment x variable when user send "1". But for new user x already not null. Please help me to fix it. 

import StringIO
import json
import logging
import random
import urllib
import urllib2

# for sending images
from PIL import Image
import multipart

# standard app engine imports
from google.appengine.api import urlfetch
from google.appengine.ext import ndb
import webapp2

TOKEN = 'TOKEN HERE'

BASE_URL = 'https://api.telegram.org/bot' + TOKEN + '/'

x = 0

# ================================

class EnableStatus(ndb.Model):
# key name: str(chat_id)
enabled = ndb.BooleanProperty(indexed=False, default=False)


# ================================

def setEnabled(chat_id, yes):
es = EnableStatus.get_or_insert(str(chat_id))
es.enabled = yes
es.put()

def getEnabled(chat_id):
es = EnableStatus.get_by_id(str(chat_id))
if es:
    return es.enabled
return False


# ================================

class MeHandler(webapp2.RequestHandler):
def get(self):
    urlfetch.set_default_fetch_deadline(60)
    self.response.write(json.dumps(json.load(urllib2.urlopen(BASE_URL + 'getMe'))))


class GetUpdatesHandler(webapp2.RequestHandler):
def get(self):
    urlfetch.set_default_fetch_deadline(60)
    self.response.write(json.dumps(json.load(urllib2.urlopen(BASE_URL + 'getUpdates'))))


class SetWebhookHandler(webapp2.RequestHandler):
def get(self):
    urlfetch.set_default_fetch_deadline(60)
    url = self.request.get('url')
    if url:
        self.response.write(json.dumps(json.load(urllib2.urlopen(BASE_URL + 'setWebhook', urllib.urlencode({'url': url})))))


class WebhookHandler(webapp2.RequestHandler):
def post(self):
    urlfetch.set_default_fetch_deadline(60)
    body = json.loads(self.request.body)
    logging.info('request body:')
    logging.info(body)
    self.response.write(json.dumps(body))

    update_id = body['update_id']
    message = body['message']
    message_id = message.get('message_id')
    date = message.get('date')
    text = message.get('text')
    fr = message.get('from')
    chat = message['chat']
    chat_id = chat['id']

    if not text:
        logging.info('no text')
        return
    def reply(msg=None, img=None):
        if msg:
            resp = urllib2.urlopen(BASE_URL + 'sendMessage', urllib.urlencode({
                'chat_id': str(chat_id),
                'text': msg.encode('utf-8'),
                'disable_web_page_preview': 'true',
                #'reply_to_message_id': str(message_id),
            })).read()
        elif img:
            resp = multipart.post_multipart(BASE_URL + 'sendPhoto', [
                ('chat_id', str(chat_id)),
                #('reply_to_message_id', str(message_id)),
            ], [
                ('photo', 'image.jpg', img),
            ])
        else:
            logging.error('no msg or img specified')
            resp = None

        logging.info('send response:')
        logging.info(resp)

    if text.startswith('/'):
        if text == '/start':
            reply('Bot start')
            setEnabled(chat_id, True)
        elif text == '/stop':
            reply('Bot disabled')
            setEnabled(chat_id, False)
        else:
            reply('What command?')

    elif '1' in text:
        global x
        x = x + 1
        reply(str(x))
    else:
        if getEnabled(chat_id):
            reply('?')
        else:
            logging.info('not enabled for chat_id {}'.format(chat_id))


app = webapp2.WSGIApplication([
('/me', MeHandler),
('/updates', GetUpdatesHandler),
('/set_webhook', SetWebhookHandler),
('/webhook', WebhookHandler),
], debug=True)
Dennix
  • 109
  • 1
  • 8
  • The `x` variable is per user or must be shared globally? PS. I know you declared it `global` but I want to be sure about your intentions. – unnikked Mar 24 '16 at 13:08
  • I want to drop x to every new user. But I dont know how to do it correctly. – Dennix Mar 24 '16 at 13:27
  • Every user who connect to bot have to start work with script like no other users, and the script is the first time for him with initial state e.g. x=0. – Dennix Mar 24 '16 at 13:39
  • x should be stored per user, not as a global variable. Global variable is shared across all requests to a particular instance – marcadian Mar 25 '16 at 00:18
  • I understand it but I dont know how exactly to do it – Dennix Mar 25 '16 at 01:16

1 Answers1

0

Instead of variable x, you may want to use python dictionary like this -

x = {}

and then use it like this -

key = str(chat_id)
if key in x:
  x[key] += 1
else:
  x[key] = 1
lamirap
  • 510
  • 1
  • 3
  • 8