使模块使用主程序中定义的函数和变量

时间:2014-05-16 18:57:57

标签: python module irc

我为我的网络制作了一个IRC机器人。为了使代码更清晰,我在模块中定义函数。所有这些模块都在名为" plugins"的文件夹中。一个模块调用函数 sendMsg 并失败,因为它试图运行主程序中定义的函数。我还想让这个模块能够在程序启动后访问主程序中定义的变量。

import socket
import time
import re

from plugins.say import *

host = "irc.somenetwork.net"
port = 6667
nick = "ircbot"
channels = "##bottesting"
s = socket.socket()

def connect():
   s.connect((host, port))
   s.send("NICK %s\r\n" % nick)
   s.send("USER %s %s nul :%s\r\n" % (nick, nick, nick))
   time.sleep(3)
   s.send("JOIN %s\r\n" % channels)
   time.sleep(3)

def sendMsg(chan, msgSend):
   s.send("PRIVMSG %s :%s\r\n" % (chan,msgSend))
#   print "Sending message \"%s\" to channel/user \"%s\"" % (msgSend, chan)

def quitServ(quitMsg="m8bot"):
   s.send("QUIT %s\r\n" % quitMsg)

connect()

while 1:
   msgRaw = s.recv(1024)
   print msgRaw
   if msgRaw == "":
      break
   if "PING" in msgRaw:
      print "Pong!"
      PONG = msgRaw.split(' ')
      PONG[0] = PONG[0].replace('PING','PONG')
      PONG = ' '.join(PONG)
      s.send("%s\r\n" % PONG)
   if "PRIVMSG" in msgRaw:
#      print "PRIVMSG detected"
      user = ''.join(re.compile("(?<=:).{0,}(?=.{0,}!)").findall(msgRaw.split(' ')[0]))
      channel = msgRaw.split(' ')[2]
      command = (' '.join(msgRaw.split(' ')[3:]).replace(":","",1)).split(' ')[0]
      msg = ''.join((' '.join(msgRaw.split(' ')[3:]).replace(":","",1)).split(' ')[1:]).replace("\r\n","")

      if not "#" in channel:
         channel = user
      print "Nick: %s\nChannel: %s\nCommand: %s\nMsg: %s" % (user,channel,command,msg)
      if ".quit" in command:
         if msg:
            quitServ(str(msg))
         else:
            quitServ()
      if ".hello" in command:
#         print "Attempting to send Hello World message."
         sendMsg(channel, "Hello World!")
      if ".say" in command:
         say(msg)

quitServ()

这是我的计划。函数说()如下:

def say(msg):
   sendMsg(channel, "You said: %s" % msg)

我可以看到问题所在,但我不知道如何解决这个问题。反馈意见:))

-Nia

1 个答案:

答案 0 :(得分:0)

在say.py中尝试import __main__。 然后你的说法功能应该是这样的:

def say():
   __main__.sendMsg(channel, "You said: %s" % msg)

但这不是最好的解决方案。只需更少代码更改即可解决问题。