207 lines
8.0 KiB
Python
207 lines
8.0 KiB
Python
#!/usr/bin/env python3
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program 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 General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with this program; if not, write to the Free Software Foundation, Inc.,
|
|
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
import socket
|
|
import random
|
|
import time
|
|
import shelve
|
|
import _pickle
|
|
|
|
#dealer class
|
|
class Dealer:
|
|
#init assigns everyone a set amount of resources
|
|
def __init__(self, name):
|
|
self.name = name
|
|
#create unique save files for everyone
|
|
f = shelve.open("{}.dat".format(self.name), flag="c", writeback=True)
|
|
f["name"] = self.name
|
|
#assign drugs for first time, or load existing file
|
|
if not "inventory" in f:
|
|
f["inventory"] = {}
|
|
f["inventory"]["money"] = 500
|
|
f["inventory"]["meth"] = 10
|
|
f["inventory"]["heroin"] = 10
|
|
f["inventory"]["cocaine"] = 10
|
|
f.sync()
|
|
self.inventory = {}
|
|
self.inventory["money"] = 500
|
|
self.inventory["meth"] = 10
|
|
self.inventory["heroin"] = 10
|
|
self.inventory["cocaine"] = 10
|
|
else:
|
|
self.inventory = f["inventory"]
|
|
f.close()
|
|
|
|
#sell function, writes to save file
|
|
def sell(self, drug, price, amount):
|
|
f = shelve.open("{}.dat".format(self.name), flag="c", writeback=True)
|
|
if (int(f["inventory"][drug] >= 1)):
|
|
f["inventory"][drug] -= int(amount)
|
|
f["inventory"]["money"] += (int(amount) * int(price))
|
|
drug_respond = str(drug)
|
|
drug_respond_amount = str(f["inventory"][drug])
|
|
drug_respond_cash = str(f["inventory"]["money"])
|
|
#responds to channel, then syncs
|
|
sendmsg(channel, ("You have " + drug_respond_amount + " " + str(drug) + " left, and $" + drug_respond_cash + " left"))
|
|
f.sync()
|
|
f.close()
|
|
else:
|
|
sendmsg(channel, "Not enough drugs")
|
|
#buy function
|
|
def buy(self, drug, price, amount):
|
|
f = shelve.open("{}.dat".format(self.name), flag="c", writeback=True)
|
|
if (int((f["inventory"]["money"])) >= (int(amount) * int(price))):
|
|
f["inventory"][drug] += int(amount)
|
|
f["inventory"]["money"] -= (int(amount) * int(price))
|
|
drug_respond = str(drug)
|
|
drug_respond_amount = str(f["inventory"][drug])
|
|
drug_respond_cash = str(f["inventory"]["money"])
|
|
#responds to channel, then syncs
|
|
sendmsg(channel, ("You have " + drug_respond_amount + " " + str(drug) + " left, and $" + drug_respond_cash + " left"))
|
|
f.sync()
|
|
f.close()
|
|
else:
|
|
sendmsg(channel, "Not enough cash")
|
|
|
|
|
|
#allows user to check amounts of resources
|
|
def check_amount(self, drug):
|
|
f = shelve.open("{}.dat".format(self.name), flag="c", writeback=True)
|
|
if drug is "heroin":
|
|
return(f["inventory"]['heroin'])
|
|
f.sync()
|
|
f.close()
|
|
if drug is "cocaine":
|
|
return(self.inventory['cocaine'])
|
|
f.sync()
|
|
f.close()
|
|
if drug is "meth":
|
|
return(self.inventory['meth'])
|
|
f.sync()
|
|
f.close()
|
|
if drug is "money":
|
|
return(self.inventory['money'])
|
|
f.sync()
|
|
f.close()
|
|
else:
|
|
return("Drug not specified")
|
|
f.close()
|
|
|
|
|
|
#******** IRC SHIT *********
|
|
|
|
server = "irc.freenode.net"
|
|
channel = "#lainchan"
|
|
botnick = "drug-bot"
|
|
password = ""
|
|
#all bytes must be converted to UTF-8
|
|
def joinchan(chan): #channel join function
|
|
ircsock.send(bytes("JOIN "+ chan +"\n", "UTF-8"))
|
|
|
|
#creates sockets, assigns nicks and usernames, and identifies
|
|
ircsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
ircsock.connect((server, 6667))
|
|
ircsock.send(bytes("USER "+ botnick +" "+ botnick +" "+ botnick +" :connected\n", "UTF-8"))
|
|
ircsock.send(bytes("NICK "+ botnick +"\n", "UTF-8"))
|
|
ircsock.send(bytes("NICKSERV :IDENTIFY %s\r\n" % password, "UTF-8"))
|
|
#sleep for 2 to give server time to process
|
|
time.sleep(2)
|
|
joinchan(channel) #joins channel
|
|
readbuffer = ""
|
|
#main loop
|
|
while 1:
|
|
#assigns random drug prices
|
|
drug_price = random.randrange(5,15,1)
|
|
#function to send messages to IRC channel
|
|
def sendmsg(chan , msg):
|
|
ircsock.send(bytes("PRIVMSG "+ chan +" :"+ msg +"\n", "UTF-8"))
|
|
#creates a buffer to process the data
|
|
readbuffer = readbuffer+ircsock.recv(1024).decode("UTF-8")
|
|
#splits into lines then prints out in shell
|
|
temp = str.split(readbuffer, "\n")
|
|
print(readbuffer)
|
|
readbuffer=temp.pop( )
|
|
|
|
for line in temp:
|
|
line = str.rstrip(line)
|
|
line = str.split(line)
|
|
try:
|
|
#respond to pings
|
|
if(line[0] == "PING"):
|
|
ircsock.send(bytes("QUOTE :PONG %s\r\n" % line[1], "UTF-8"))
|
|
#if it finds .sell, it calls the function from the drug user class
|
|
if(line[3] == ":.sell"):
|
|
#get name of person who called command
|
|
sender = ""
|
|
for char in line[0]:
|
|
if(char == "!"):
|
|
break
|
|
if(char != ":"):
|
|
sender += char
|
|
#assigns player a save file
|
|
current_player = Dealer(sender)
|
|
#drug wanted
|
|
drug_query = line[5]
|
|
#amount of drug
|
|
drug_amount = line[4]
|
|
#calls the sell function from dealer class, passing in data collected above
|
|
current_player.sell(drug_query, drug_price, drug_amount)
|
|
if(line[3] == ":.buy"):
|
|
#get name of person who called command
|
|
sender = ""
|
|
for char in line[0]:
|
|
if(char == "!"):
|
|
break
|
|
if(char != ":"):
|
|
sender += char
|
|
#assigns player a save file
|
|
current_player = Dealer(sender)
|
|
#type of drug
|
|
drug_query = line[5]
|
|
#amount of drug
|
|
drug_amount = line[4]
|
|
#calls the buy function from dealer class, passing in data collected above
|
|
current_player.buy(drug_query, drug_price, drug_amount)
|
|
if(line[3] == ":.check"):
|
|
sender = ""
|
|
for char in line[0]:
|
|
if(char == "!"):
|
|
break
|
|
if(char != ":"):
|
|
sender += char
|
|
|
|
current_player = Dealer(sender)
|
|
requested_item = line[4]
|
|
#returns the value of the requested item
|
|
if requested_item:
|
|
if(requested_item == 'heroin'):
|
|
sendmsg(channel, (sender + ", you have " + str(current_player.check_amount('heroin')) + " heroin left" +"\n"))
|
|
elif(requested_item == 'meth'):
|
|
sendmsg(channel, (sender + ", you have " + str(current_player.check_amount('meth')) + " meth left" +"\n"))
|
|
elif(requested_item == 'money'):
|
|
sendmsg(channel, (sender + ", you have " + str(current_player.check_amount('money')) + " money left" +"\n" ))
|
|
elif(requested_item == 'cocaine'):
|
|
sendmsg(channel, (sender + ", you have " + str(current_player.check_amount('cocaine')) + " cocaine left" +"\n"))
|
|
else:
|
|
sendmsg(channel, ("Invalid option. Options for .check are heroin, meth, cocaine, or money"))
|
|
except:
|
|
continue
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|