def test_trash():
with session_scope():
ecf = EventFactory()
cf = ChatFunction()
new_event = ecf.next_event('MASTER/MASTER', {})
fcc = FeedCtrlConnection()
fcc.add_event(new_event)
trust_id1 = generate_random_feed_id()
new_event1 = ecf.next_event('MASTER/Trust', {'feed_id': trust_id1})
fcc.add_event(new_event1)
new_event2 = ecf.next_event('MASTER/Trash', {'feed_id': trust_id1})
fcc.add_event(new_event2)
new_event2 = ecf.next_event('MASTER/Trash', {'feed_id': trust_id1})
fcc.add_event(new_event2)
new_event2 = ecf.next_event('MASTER/Trash', {'feed_id': trust_id1})
fcc.add_event(new_event2)
result = cf.get_current_event(ecf.get_feed_id())
assert result == new_event2
def create_key(self, username=None):
if username != "" and len(username) <= 16:
ecf = EventCreationTool.EventFactory()
public_key = ecf.get_feed_id()
chat_function = ChatFunction()
first_event = ecf.first_event('chat',
chat_function.get_host_master_id())
chat_function.insert_event(first_event)
self.dictionary = {'username': username, 'public_key': public_key}
pickle.dump(self.dictionary, open(pickle_file_names[1],
"wb")) # save username and key
print("Your username has been saved:", username)
self.open_Chat()
else:
self.username_field.delete(0, 'end')
self.username_field.insert(0, 'Username must be < 17 letters')
def test_add_event_and_get_host_master_id():
with session_scope():
ecf = EventFactory()
cf = ChatFunction()
new_event = ecf.next_event('MASTER/MASTER', {})
fcc = FeedCtrlConnection()
fcc.add_event(new_event)
feed = EventFactory()
new_event = feed.next_event('chat/MASTER',
{'master_feed': ecf.get_feed_id()})
fcc.add_event(new_event)
new_event = feed.next_event('chat/whateveraction', {
'messagekey': 'hallo zusammen',
'chat_id': '1',
'timestampkey': 10
})
fcc.add_event(new_event)
new_event = feed.next_event('chat/whateveraction', {
'messagekey': 'hallo zusammen',
'chat_id': '1',
'timestampkey': 10
})
fcc.add_event(new_event)
trust_id5 = generate_random_feed_id()
new_event = ecf.next_event('MASTER/Trust', {'feed_id': trust_id5})
fcc.add_event(new_event)
trust_id5 = generate_random_feed_id()
new_event = ecf.next_event('MASTER/Name', {'name': trust_id5})
fcc.add_event(new_event)
trust_id5 = generate_random_feed_id()
new_event = ecf.next_event('MASTER/NewFeed', {
'feed_id': trust_id5,
'app_name': 'TestApp'
})
fcc.add_event(new_event)
new_event = ecf.next_event('MASTER/Radius', {'radius': 5})
fcc.add_event(new_event)
result = cf.get_full_chat('1')
logger.error(result)
result = fcc.get_host_master_id()
assert result == ecf.get_feed_id()
yellow_rect = pygame.draw.rect(
screen, yellow.colour,
(correctposition(yellow.colour, yellow.position) + (radius, radius)))
green_rect = pygame.draw.rect(screen, yellow.colour,
(correctposition(green.colour, green.position) +
(radius, radius)))
black_rect = pygame.draw.rect(screen, yellow.colour,
(correctposition(black.colour, black.position) +
(radius, radius)))
red_rect = pygame.draw.rect(screen, yellow.colour,
(correctposition(red.colour, red.position) +
(radius, radius)))
ecf = EventCreationTool.EventFactory()
public_key = ecf.get_feed_id()
chat_function = ChatFunction()
first_event = ecf.first_event('chat', chat_function.get_host_master_id())
chat_function.insert_event(first_event)
dictionary = {'role': 'free', 'public_key': public_key}
pickle.dump(dictionary, open(pickle_file_names[1], "wb"))
# Set EventFactory
chat_function = ChatFunction()
chat_function.get_current_event(chat_function.get_all_feed_ids()[1])
feed_id = dictionary['public_key']
most_recent_event = chat_function.get_current_event(feed_id)
ecf = EventCreationTool.EventFactory(
most_recent_event) # so that the same ID is used
def test_get_chat_event():
with session_scope():
with LogCapture() as log_cap:
ecf = EventFactory()
cf = ChatFunction()
new_event = ecf.next_event('MASTER/MASTER', {})
cf.insert_event(new_event)
feed = EventFactory()
feed2 = EventFactory()
feed3 = EventFactory()
new_event = feed.next_event('chat/MASTER',
{'master_feed': ecf.get_feed_id()})
cf.insert_event(new_event)
new_event = feed2.next_event('chat/MASTER',
{'master_feed': ecf.get_feed_id()})
cf.insert_event(new_event)
new_event = feed3.next_event('chat/MASTER',
{'master_feed': ecf.get_feed_id()})
cf.insert_event(new_event)
new_event = feed.next_event('chat/whateveraction', {
'messagekey': 'hallo zusammen',
'chat_id': '1',
'timestampkey': 10
})
cf.insert_event(new_event)
new_event = feed.next_event('chat/whateveraction', {
'messagekey': 'wie gehts?',
'chat_id': '1',
'timestampkey': 20
})
cf.insert_event(new_event)
new_event = feed.next_event(
'chat/whateveraction', {
'messagekey': 'schönes Wetter heute',
'chat_id': '1',
'timestampkey': 30
})
cf.insert_event(new_event)
print(log_cap)
q = cf.get_chat_since(15, '1')
assert q[0][0] == 'wie gehts?'
assert q[1][0] == 'schönes Wetter heute'
t = cf.get_full_chat('1')
assert t[0][0] == 'hallo zusammen'
assert t[1][0] == 'wie gehts?'
assert t[2][0] == 'schönes Wetter heute'
assert True
def __init__(self, master=None):
Frame.__init__(self, master)
master.title("BAC net")
# Set Variables
self.msg = StringVar()
self.dictionary = pickle.load(open(pickle_file_names[1], "rb"))
self.username = self.dictionary['username']
self.feed_id = self.dictionary['public_key']
self.partner = list(
) # empty partner as default... tuple: name, chatID
self.partner.append("")
self.partner.append("")
self.entry_field = StringVar()
self.alreadyUpdatedIndex = 0
self.time = datetime.datetime.now()
self.lastMessage = list()
self.chat_function = ChatFunction()
# Set EventFactory
x = self.chat_function.get_current_event(
self.chat_function.get_all_feed_ids()[1])
most_recent_event = self.chat_function.get_current_event(self.feed_id)
self.ecf = EventCreationTool.EventFactory(
most_recent_event) # damit wieder gleiche Id benutzt wird
# set self.personList:
try: # Try to load the Variable: if it works, we know we can load it and assign it
x = pickle.load(
open(pickle_file_names[0], "rb")
) # if it fails then we know we have not populated our pkl file yet
self.person_list = pickle.load(open(pickle_file_names[0], "rb"))
except: # if we can't load the variable, we need to assign and dump it first
List = list()
pickle.dump(List, open(pickle_file_names[0],
"wb")) # create an empty object
self.person_list = pickle.load(open(
pickle_file_names[0], "rb")) # loading and setting it now
# Configure Application layout
self.master.configure(height=24, width=50)
self.tabFrame = Frame(self.master)
self.tabFrame.pack(side='left', fill='both', expand=True)
self.chatFrame = Frame(self.master)
self.chatFrame.pack(side='right', fill='both', expand=True)
# TabFrame new message area
self.tabTopBar = Frame(self.tabFrame)
self.tabTopBar.config(bg="#ededed")
# Configure Frames
self.topFrameChat = Frame(self.chatFrame)
self.topFrameChat.pack(fill='both', expand=True)
self.middleFrameChat = Frame(self.chatFrame)
self.middleFrameChat.pack(fill='both', expand=True)
self.bottomFrameChat = Frame(self.chatFrame)
self.bottomFrameChat.pack(fill='both', expand=True)
# Place Widgets
self.createWidgets()
# now we can add the partner(s) into ListBox3
self.addPartners()
# Configure Subframes
self.rowconfigure(1, weight=1)
self.columnconfigure(1, weight=1)
self.master.rowconfigure(1, weight=1)
self.master.columnconfigure(1, weight=1)
self.tabFrame.rowconfigure(1, weight=1)
self.chatFrame.columnconfigure(1, weight=1)
self.topFrameChat.rowconfigure(1, weight=1)
self.topFrameChat.columnconfigure(1, weight=1)
self.middleFrameChat.rowconfigure(1, weight=1)
self.middleFrameChat.columnconfigure(1, weight=1)
self.bottomFrameChat.rowconfigure(1, weight=1)
self.bottomFrameChat.columnconfigure(1, weight=1)
self.tabTopBar.rowconfigure(1, weight=1)
self.tabTopBar.columnconfigure(1, weight=1)
# Placement
self.username_label.grid(row=0, column=0, columnspan=3, sticky="ew")
self.listBox1.grid(row=1, column=0, sticky="nsew")
self.listBox2.grid(row=1, column=1, sticky="nsew")
self.text_field.grid(row=2, column=0, columnspan=3, sticky="ew")
self.send_button.grid(row=3, column=0, columnspan=3, sticky="ew")
self.update_button.grid(row=4, column=0, columnspan=3, sticky="ew")
self.tabTopBar.grid(row=0, column=0, columnspan=3, sticky="nsew")
self.create_Button.grid(row=0, column=1, sticky="ew")
self.join_Button.grid(row=0, column=2, sticky="ew")
self.listBox3.grid(row=1, column=0, sticky="nsew")
class Chat(Frame):
def __init__(self, master=None):
Frame.__init__(self, master)
master.title("BAC net")
# Set Variables
self.msg = StringVar()
self.dictionary = pickle.load(open(pickle_file_names[1], "rb"))
self.username = self.dictionary['username']
self.feed_id = self.dictionary['public_key']
self.partner = list(
) # empty partner as default... tuple: name, chatID
self.partner.append("")
self.partner.append("")
self.entry_field = StringVar()
self.alreadyUpdatedIndex = 0
self.time = datetime.datetime.now()
self.lastMessage = list()
self.chat_function = ChatFunction()
# Set EventFactory
x = self.chat_function.get_current_event(
self.chat_function.get_all_feed_ids()[1])
most_recent_event = self.chat_function.get_current_event(self.feed_id)
self.ecf = EventCreationTool.EventFactory(
most_recent_event) # damit wieder gleiche Id benutzt wird
# set self.personList:
try: # Try to load the Variable: if it works, we know we can load it and assign it
x = pickle.load(
open(pickle_file_names[0], "rb")
) # if it fails then we know we have not populated our pkl file yet
self.person_list = pickle.load(open(pickle_file_names[0], "rb"))
except: # if we can't load the variable, we need to assign and dump it first
List = list()
pickle.dump(List, open(pickle_file_names[0],
"wb")) # create an empty object
self.person_list = pickle.load(open(
pickle_file_names[0], "rb")) # loading and setting it now
# Configure Application layout
self.master.configure(height=24, width=50)
self.tabFrame = Frame(self.master)
self.tabFrame.pack(side='left', fill='both', expand=True)
self.chatFrame = Frame(self.master)
self.chatFrame.pack(side='right', fill='both', expand=True)
# TabFrame new message area
self.tabTopBar = Frame(self.tabFrame)
self.tabTopBar.config(bg="#ededed")
# Configure Frames
self.topFrameChat = Frame(self.chatFrame)
self.topFrameChat.pack(fill='both', expand=True)
self.middleFrameChat = Frame(self.chatFrame)
self.middleFrameChat.pack(fill='both', expand=True)
self.bottomFrameChat = Frame(self.chatFrame)
self.bottomFrameChat.pack(fill='both', expand=True)
# Place Widgets
self.createWidgets()
# now we can add the partner(s) into ListBox3
self.addPartners()
# Configure Subframes
self.rowconfigure(1, weight=1)
self.columnconfigure(1, weight=1)
self.master.rowconfigure(1, weight=1)
self.master.columnconfigure(1, weight=1)
self.tabFrame.rowconfigure(1, weight=1)
self.chatFrame.columnconfigure(1, weight=1)
self.topFrameChat.rowconfigure(1, weight=1)
self.topFrameChat.columnconfigure(1, weight=1)
self.middleFrameChat.rowconfigure(1, weight=1)
self.middleFrameChat.columnconfigure(1, weight=1)
self.bottomFrameChat.rowconfigure(1, weight=1)
self.bottomFrameChat.columnconfigure(1, weight=1)
self.tabTopBar.rowconfigure(1, weight=1)
self.tabTopBar.columnconfigure(1, weight=1)
# Placement
self.username_label.grid(row=0, column=0, columnspan=3, sticky="ew")
self.listBox1.grid(row=1, column=0, sticky="nsew")
self.listBox2.grid(row=1, column=1, sticky="nsew")
self.text_field.grid(row=2, column=0, columnspan=3, sticky="ew")
self.send_button.grid(row=3, column=0, columnspan=3, sticky="ew")
self.update_button.grid(row=4, column=0, columnspan=3, sticky="ew")
self.tabTopBar.grid(row=0, column=0, columnspan=3, sticky="nsew")
self.create_Button.grid(row=0, column=1, sticky="ew")
self.join_Button.grid(row=0, column=2, sticky="ew")
self.listBox3.grid(row=1, column=0, sticky="nsew")
def createWidgets(self):
print("connected as: \"" + self.username + "\"")
# window title: shows by what username you are connected to the application
self.master.title("BAC net - " + self.username.upper())
# Chat: title bar: shows to who you are currently writing
self.username_label = Label(self.topFrameChat, text="")
self.username_label.configure(bg="#ededed", font=('HelveticaNeue', 14))
# Chat: chatfield
self.scrollbar = Scrollbar(self.middleFrameChat, orient="vertical")
self.scrollbar.config(command=self.yview)
self.listBox1 = Listbox(self.middleFrameChat,
height=30,
width=25,
yscrollcommand=self.scroll1)
self.listBox1.configure(bg='#e3dbd4', font=('HelveticaNeue', 10))
self.listBox1.bind("<<ListboxSelect>>", lambda x: self.open_file1())
self.listBox2 = Listbox(self.middleFrameChat,
height=30,
width=25,
yscrollcommand=self.scroll2)
self.listBox2.configure(bg='#e3dbd4', font=('HelveticaNeue', 10))
self.listBox2.bind("<<ListboxSelect>>", lambda x: self.open_file2())
# ---Tab Section---
self.tabScrollbar = Scrollbar(self.tabFrame, orient="vertical")
self.listBox3 = Listbox(self.tabFrame,
height=23,
width=25,
yscrollcommand=self.tabScrollbar.set,
selectbackground="#ededed")
self.tabScrollbar.config(command=self.listBox3.yview)
self.listBox3.configure(bg='white', font=('HelveticaNeue', 15, 'bold'))
self.listBox3.bind("<<ListboxSelect>>", lambda x: self.loadChat())
# Chat: actions
self.text_field = Entry(self.bottomFrameChat,
textvariable=self.msg,
font=('HelveticaNeue', 10))
self.text_field.configure(bg='#f0f0f0')
self.text_field.bind(
'<Return>',
lambda event: self.check_and_save(self.msg.get(), self.partner[1]))
self.text_field.bind("<Button-1>",
lambda event: self.clear_on_entry("text_field"))
self.send_button = Button(self.bottomFrameChat,
text='Send',
command=lambda: self.check_and_save(
self.msg.get(), self.partner[1]),
bg="#25D366",
activebackground="#21B858",
font=('HelveticaNeue', 10))
self.update_button = Button(
self.bottomFrameChat,
text='Update',
command=lambda: self.loadChat(self.partner),
bg="white",
font=('HelveticaNeue', 10))
self.create_Button = Button(
self.tabTopBar,
text=" create ",
command=lambda: self.saveTypeAndSwitchState("create"),
bg="#25D366",
activebackground="#21B858",
font=('HelveticaNeue', 11)) # green
self.join_Button = Button(
self.tabTopBar,
text=" join ",
command=lambda: self.saveTypeAndSwitchState("join"),
bg="#34B7F1",
activebackground="#0f9bd7",
font=('HelveticaNeue', 11)) # blue
self.privateChat_Button = Button(
self.tabTopBar,
text="Private Chat",
command=lambda: self.saveTypeAndSwitchState("private Chat"),
bg="#25D366",
activebackground="#21B858",
font=('HelveticaNeue', 11)) # lightgreen
self.group_Button = Button(
self.tabTopBar,
text=" Group ",
command=lambda: self.saveTypeAndSwitchState("group"),
bg="#34B7F1",
activebackground="#0f9bd7",
font=('HelveticaNeue', 11)) # lightblue
self.confirm_Button = Button(
self.tabTopBar,
text="OK",
command=lambda: self.saveTypeAndSwitchState("confirm"),
bg="#25D366",
activebackground="#21B858",
font=('HelveticaNeue', 11)) # green
self.back_Button = Button(
self.tabTopBar,
text="back",
command=lambda: self.saveTypeAndSwitchState("back"),
bg="white",
font=('HelveticaNeue', 11),
highlightcolor="white") # green
self.button_state = 0
self.ButtonType = ""
self.ButtonTask = ""
self.BackTask = ""
self.id_field = Entry(self.tabTopBar,
textvariable=self.entry_field,
state=NORMAL) # readonly state
self.id_field.bind("<Button-1>",
lambda event: self.clear_on_entry("id_field"))
self.id_field.bind('<Return>', lambda event: self.switchState())
# --------- GRAPHICAL HELP FUNCTIONS ----------
def scroll1(
self, *args
): # when the user scrolls the listBox1, then this method ensures the position of listBox2 is synced
if self.listBox2.yview() != self.listBox1.yview(
): # when the listBox2 is out of sync...
self.listBox2.yview_moveto(
args[0]) # ...adjust the yview of listBox2
# self.scrollbar.set(*args) # sync the scrollbar with the mouse wheel position
def scroll2(
self, *args
): # when the user scrolls the listBox2, then this method ensures the position of listBox1 is synced
if self.listBox1.yview() != self.listBox2.yview(
): # when the listBox1 is out of sync...
self.listBox1.yview_moveto(
args[0]) # ...adjust the yview of listBox1
# self.scrollbar.set(*args) # sync the scrollbar with the mouse wheel position
def yview(
self, *args
): # self configured yview command. the only job is to trigger both sync methods from abouve
self.listBox1.yview(*args)
self.listBox2.yview(*args)
def add(self, chatID):
self.listBox1.delete(0, 'end')
self.listBox2.delete(0, 'end')
self.time = datetime.datetime.now()
timezone = 7200 # +2h, so we have the european time in the most simplistic way
chat = self.chat_function.get_full_chat(chatID)
chat_type = chat[0][0].split("#split:#")[
3] # get the type of the chat (private or group)
for i in range(1, len(chat)):
chat_message = chat[i][0].split(
"#split:#"
) # a chat-message is like: username#split:#message, so we need to split this two
partner_username = chat_message[0] # from who is the message
message = chat_message[1] # the real content / message
additional_msg = chat_message[2]
if len(chat_message) == 4:
if chat_type == "private" and self.partner[0] == self.partner[
1] and additional_msg == "member":
for i in range(len(self.person_list)):
if self.partner[1] == self.person_list[i][1]:
self.person_list[i][
0] = partner_username # the creator of the group gets the name of his partner for the first time
self.partner[0] = partner_username
self.username_label.config(
text=TextWrapper.shorten_name(
self.partner[0], 34))
else: # length of message == 3
if partner_username != self.username: # message from the partner(s)
# print the message with white background:
if chat_type == "group":
self.listBox1.insert('end', partner_username + ":")
try:
self.listBox1.itemconfig(
'end',
bg='white',
foreground=Colorize.name_to_color(
partner_username))
except:
self.listBox1.itemconfig('end',
bg='white',
foreground='#00a86b')
self.listBox2.insert('end', "")
if additional_msg[0:3] == "pdf":
self.listBox1.insert(
'end', "Click to open pdf. (" + str(i) + ")")
self.listBox1.itemconfig('end', bg='white')
self.listBox2.insert('end', "")
elif additional_msg[0:3] == "img":
self.listBox1.insert(
'end', "Click to open image. (" + str(i) + ")")
self.listBox1.itemconfig('end', bg='white')
self.listBox2.insert('end', "")
else:
messages = TextWrapper.textWrap(message, 0)
for i in range(len(messages)):
self.listBox1.insert('end', messages[i])
self.listBox1.itemconfig('end', bg='white')
self.listBox2.insert('end', '')
self.listBox1.insert(
'end', "{:<22}{:>16}".format(
"",
time.strftime("%H:%M %d.%m.%Y",
time.gmtime(chat[i][1] + timezone))))
self.listBox1.itemconfig('end',
bg='white',
foreground="lightgrey")
self.listBox2.insert('end', "")
self.listBox1.yview(END)
self.listBox2.yview(END)
self.listBox1.insert('end',
'') # some space to enhance appeal
self.listBox2.insert('end', '')
else: # username = self.username: message from the user
# print the message with green background:
if additional_msg[0:3] == "pdf":
self.listBox2.insert(
'end', "Click to open PDF. (" + str(i) + ")")
self.listBox2.itemconfig('end', bg='#dafac9')
self.listBox1.insert('end', '')
elif additional_msg[0:3] == "img":
self.listBox2.insert(
'end', "Click to open image. (" + str(i) + ")")
self.listBox2.itemconfig('end', bg='#dafac9')
self.listBox1.insert('end', '')
else: # == msg
messages = TextWrapper.textWrap(message, 0)
for i in range(len(messages)):
self.listBox2.insert('end', messages[i])
self.listBox2.itemconfig('end', bg='#dafac9')
self.listBox1.insert('end', '')
self.listBox2.insert(
'end', "{:<22}{:>16}".format(
"",
time.strftime("%H:%M %d.%m.%Y",
time.gmtime(chat[i][1] + timezone))))
self.listBox2.itemconfig('end',
bg='#dafac9',
foreground="lightgrey")
self.listBox1.insert('end', '')
self.listBox2.yview(END)
self.listBox1.yview(END)
self.listBox2.insert('end',
'') # some space to enhance appeal
self.listBox1.insert('end', '')
def updateContent(self, chatID):
self.add(chatID)
self.addPartners()
def loadChat(
self,
chat=None
): # when the user clicks on Person in listBox3 this function is called and loads the correct chat and sets up everything needed to start the communication
if not chat:
try:
selection = self.listBox3.curselection()[
0] # this gives an int value: first element = 0
except IndexError:
return
if selection or selection == 0: # if something is selected
self.partner[0] = self.person_list[int(selection)][0]
self.partner[1] = self.person_list[int(selection)][1]
else:
if self.partner[0] != chat[0]:
self.partner[0] = chat[0]
self.partner[1] = chat[1]
for i in range(len(self.person_list)):
if self.person_list[i][1] == self.partner[1]:
self.person_list[i][2] = time.time()
chat_type = self.chat_function.get_full_chat(
self.partner[1])[0][0].split("#split:#")[
3] # get the type of the chat (private or group)
if chat_type == "private":
self.username_label.config(
text=TextWrapper.shorten_name(self.partner[0], 34))
else: # chat_type == "group"
self.username_label.config(
text=TextWrapper.shorten_name(self.partner[0], 27) + " (" +
self.partner[1] + ")")
self.updateContent(self.partner[1])
def check_for_new_messages(self, person_nr):
length = len(
self.chat_function.get_chat_since(self.person_list[person_nr][2],
self.person_list[person_nr][1]))
unread_messages = length - self.count_users_in_chat(
self.person_list[person_nr][1], self.person_list[person_nr][2])
return unread_messages
def addPartners(self):
if len(self.person_list) != 0:
self.listBox3.delete(0, 'end')
self.person_list.sort(
key=lambda x: self.chat_function.get_full_chat(x[1])[-1][1],
reverse=True)
for i in range(len(self.person_list)):
counter = self.check_for_new_messages(i)
if self.partner[1] == self.person_list[i][1]:
counter = 0
self.listBox3.insert(
'end',
TextWrapper.mergeNameCounter(self.person_list[i][0],
counter))
if counter != 0:
self.listBox3.itemconfig(i, bg='#dfd')
else:
self.listBox3.itemconfig(i, bg="white")
def saveTypeAndSwitchState(self, Type):
if Type == 'back':
self.BackTask = Type
#self.button_state = (self.button_state - 1) % 3
elif self.button_state == 0: #Type is 'create' or 'join':
self.ButtonType = Type
if Type == 'join':
self.privateChat_Button.grid_remove()
self.group_Button.grid_remove()
self.join_Button.grid_remove()
self.id_field.insert(0, "Enter ID here")
self.confirm_Button.grid(row=0, column=1, sticky="ew")
self.id_field.grid(row=0, column=0, sticky="ew")
self.id_field.delete(0, END)
self.ButtonTask == ""
self.back_Button.grid(row=0, column=3, sticky="e")
self.button_state = 2
elif self.button_state == 1: #Type is 'private Chat' or 'group':
self.ButtonTask = Type
elif self.button_state == 2:
self.switchState(Type)
return
self.switchState()
def switchState(self,
Type=None): # creates a menu like behaviour of the buttons
# STATE = 0
if self.button_state == 0: # after first click (we now know the Button Type ('create' or 'join'))
self.create_Button.grid_remove()
self.join_Button.grid_remove()
self.privateChat_Button.grid(row=0, column=1, sticky="ew")
self.group_Button.grid(row=0, column=2, sticky="ew")
self.back_Button.grid(
row=0, column=3,
sticky="e") # add back button for second state
self.button_state = 1 # only advance by one state when back button was not activated
# STATE = 1
elif self.button_state == 1: # after second click (we now know the Button Task ('private Chat' or 'group'))
if self.BackTask != 'back':
self.privateChat_Button.grid_remove()
self.group_Button.grid_remove()
self.confirm_Button.grid(row=0, column=1, sticky="ew")
self.id_field.grid(row=0, column=0, sticky="ew")
self.id_field.delete(0, END)
if self.ButtonTask == 'group' and self.ButtonType == 'create':
self.id_field.insert(0, "Enter Group name here")
elif self.ButtonTask == 'private Chat' and self.ButtonType == 'create':
self.id_field.config(state='readonly')
self.entry_field.set(self.generate_ID())
self.button_state = 2 # only advance by one state when back button was not activated
else: # we need to reset by one state
self.BackTask = ""
self.ButtonType = ""
self.privateChat_Button.grid_remove()
self.group_Button.grid_remove()
self.back_Button.grid_remove()
self.create_Button.grid(row=0, column=1, sticky="nsew")
self.join_Button.grid(row=0, column=3, sticky="nsew")
self.button_state = 0
# STATE = 2
else: # self.button_state == 2: #after third click (reset to state = 0)
error_type = 'None'
if self.BackTask != 'back':
if self.ButtonType == 'join':
if self.is_joinable(self.id_field.get()):
self.join_chat(self.id_field.get())
elif Type == 'confirm':
self.id_field.delete(0, END)
error_type = "try again: enter ID"
else:
self.id_field.delete(0, END)
error_type = "enter ID here"
elif self.ButtonTask == 'group' and self.ButtonType == 'create':
if len(self.id_field.get()) <= 16 and self.id_field.get(
) != "" and self.id_field.get() != " ":
self.create_chat(self.generate_ID(),
self.id_field.get())
else:
error_type = "Name must be < 17"
elif self.ButtonTask == 'private Chat' and self.ButtonType == 'create':
self.create_chat(self.id_field.get())
if error_type == 'None':
self.BackTask = ""
self.ButtonType = ""
self.id_field.grid_remove()
self.confirm_Button.grid_remove()
self.id_field.config(state=NORMAL)
self.back_Button.grid_remove()
self.create_Button.grid(row=0, column=1, sticky="ew")
self.join_Button.grid(row=0, column=2, sticky="ew")
self.button_state = 0 # only advance by one state when back button was not activated
else: # 'error' occured
self.id_field.delete(0, END)
self.id_field.insert(0, error_type)
error_type = 'None'
else: # we need to reset by one state
self.BackTask = ""
self.id_field.delete(0, END)
self.id_field.grid_remove()
self.id_field.config(state=NORMAL)
self.confirm_Button.grid_remove()
if self.ButtonTask == 'group' or self.ButtonTask == 'private Chat':
self.ButtonTask = ""
self.privateChat_Button.grid(row=0, column=1, sticky="ew")
self.group_Button.grid(row=0, column=2, sticky="ew")
self.button_state = 1
else: # join was previously selected
self.ButtonTask = ""
self.back_Button.grid_remove()
self.create_Button.grid(row=0, column=1, sticky="ew")
self.join_Button.grid(row=0, column=2, sticky="ew")
self.button_state = 0
if self.partner[1] != "":
self.loadChat(self.partner)
def clear_on_entry(self, field):
if field == 'id_field':
if self.id_field.get(
) == "Enter Group ID here" or self.id_field.get(
) == "try again: enter ID" or self.id_field.get(
) == "Name must be < 17" or self.id_field.get(
) == "enter ID here" or self.id_field.get(
) == "Enter Group name here":
self.id_field.delete(0, 'end')
elif field == 'text_field':
if self.text_field.get(
) == "Sorry, given path could not be loaded, please try again!" or self.text_field.get(
) == "Sorry, file must not be < 100 KB" or self.text_field.get(
) == "Sorry, file does not exist, please try again!" or self.text_field.get(
) == "Sorry, given path does not exist, please try again!":
self.text_field.delete(0, 'end')
if self.partner[1] != "":
self.loadChat(self.partner)
# -------------- -------------- --------------
# -------------- HELP FUNCTIONS --------------
def check_and_save(self, message, chat_id):
if self.partner[1] != "":
count = 0 # check if the message consists of only spaces
for i in range(len(message)):
if message[
i] != " " and count == 0: # check how many chars are not spaces. if we found one that is not a space, we can continue
count += 1
if message == "" or count == 0 or not self.person_list or not self.partner:
pass
elif message[0:5] == 'img: ' or message[0:5] == 'pdf: ':
file_type = message[0:3]
file_path = message[5:]
if os.path.isdir(file_path[0:file_path.rfind('/')]
): # check if path is a valid directory
if os.path.isfile(file_path): # check if path has a file
file_string = self.encode_file(file_path)
file_name = file_path[file_path.rfind('/') + 1:]
length = len(file_string)
kilobyte = 1000 # a thousand chars fit into one kilobyte
if length > 100 * kilobyte: # file too big
self.text_field.delete(0, 'end')
self.text_field.insert(
0, "Sorry, file must not be > 100 KB")
elif length > kilobyte: # if more than one KiloByte
partition_len = int(math.ceil(length / kilobyte))
for i in range(partition_len):
if i == int(partition_len
) - 1: # We arrived at last part
self.save(
file_string[i * kilobyte:] +
"#split:#" + file_type +
str(partition_len) + "_" + file_name,
chat_id)
else:
self.save(
file_string[i * kilobyte:(i + 1) *
kilobyte] +
"#split:#filepart#split:#filepart",
chat_id)
else: # length <= 1000
self.save(
file_string + "#split:#" + file_type +
file_name, chat_id)
else: # file not found
self.text_field.delete(0, 'end')
self.text_field.insert(
0, "Sorry, file does not exist, please try again!")
else: # directory not found
self.text_field.delete(0, 'end')
self.text_field.insert(
0,
"Sorry, given path does not exist, please try again!")
else: # normal message recognized
self.save(message + "#split:#msg", chat_id)
else:
self.text_field.delete(0, 'end')
def save(self, message, chatID):
to_save = self.username + "#split:#" + message
new_event = self.ecf.next_event('chat/saveMessage', {
'messagekey': to_save,
'chat_id': chatID,
'timestampkey': time.time()
})
self.chat_function.insert_chat_msg(new_event)
if self.partner[1] != "":
self.loadChat(
self.partner
) #updating chat, so the send message is also added in the listbox
self.text_field.delete(0, 'end')
def create_chat(self, ID, name=None):
if not name:
self.person_list.append([ID, ID, 0])
self.partner[0] = ID
self.partner[1] = ID
self.save(self.username + "#split:#member#split:#private", ID)
else: #group was created
self.person_list.append([name, ID,
0]) # it's a group so we do know the name
self.partner[0] = name
self.partner[1] = ID
self.save(name + "#split:#member#split:#group", ID)
self.addPartners()
self.loadChat(self.partner)
def join_chat(self, ID):
if self.is_joinable(ID):
self.person_list.append([ID, ID, 0])
partnerName = self.chat_function.get_full_chat(ID)[0][0].split(
"#split:#"
)[1] # taking the name of the partner for the frome the first message
self.person_list[-1][
0] = partnerName # index -1 is the index of the last list-element
self.partner[0] = partnerName
self.partner[1] = ID
self.save(self.username + "#split:#member#split:#chat", ID)
self.addPartners()
self.loadChat(self.partner)
def count_users_in_chat(self, chatID, since=None):
counter = 0
if not since:
chat = self.chat_function.get_full_chat(chatID)
else:
chat = self.chat_function.get_chat_since(since, chatID)
for i in range(len(chat)):
if len(chat[i][0].split("#split:#")) == 4:
counter += 1
return counter
def is_joinable(self, chatID):
if len(chatID) != 6:
return False # invalid input
else:
chat = self.chat_function.get_full_chat(chatID)
if len(chat) == 0:
return False # the chat isn't even created
if chat[0][0].split(
"#split:#")[3] == "private" and self.count_users_in_chat(
chatID) == 2: # max. member for private chat: 2
return False
for i in range(len(self.person_list)
): # So the person cannot write with himself
if self.person_list[i][
1] == chatID: # you are already in this chat
return False
return True
def assemble_file_parts(self, parts):
assembled_content = ""
for i in range(len(parts)):
assembled_content = assembled_content + parts[i]
return assembled_content
def open_file1(self):
global switch
try:
selection = self.listBox1.curselection()[
0] # this gives an int value: first element = 0
except IndexError:
return
if selection or selection == 0:
item = self.listBox1.get(selection)
if len(item) >= 21 and (
item[0:19] == "Click to open PDF. "
or item[0:21] == "Click to open image. "
): # the content should be "Click to open the file. (index)"
index = int(item[item.find("(") +
1:item.find(")")]) # getting the index
messages = self.chat_function.get_full_chat(self.partner[1])
part_as_string = messages[index][0].split("#split:#")
sender_of_file = part_as_string[0]
content_of_file = list()
content_of_file.append(part_as_string[1]) #last part
name_of_file = part_as_string[2]
counter = int(name_of_file[3:name_of_file.find("_")]) - 1
while counter != 0:
index -= 1
part_as_string = messages[index][0].split("#split:#")
if len(part_as_string
) == 4 and part_as_string[0] == sender_of_file:
content_of_file.append(part_as_string[1])
counter -= 1
type_of_file = ""
if name_of_file[0:3] == "pdf":
type_of_file = "pdf"
elif name_of_file[0:3] == "img":
type_of_file = "img"
content_of_file.reverse()
assembled_content = self.assemble_file_parts(content_of_file)
self.open_file(assembled_content,
name_of_file[name_of_file.find("_") + 1:],
type_of_file)
def open_file2(self):
try:
selection = self.listBox2.curselection()[
0] # this gives an int value: first element = 0
except IndexError:
return
if selection or selection == 0:
item = self.listBox2.get(selection)
if len(item) >= 21 and (
item[0:19] == "Click to open PDF. "
or item[0:21] == "Click to open image. "
): # the content should be "Click to open the file. (index)"
index = int(item[item.find("(") +
1:item.find(")")]) # getting the index
messages = self.chat_function.get_full_chat(self.partner[1])
part_as_string = messages[index][0].split("#split:#")
sender_of_file = part_as_string[0]
content_of_file = list()
content_of_file.append(part_as_string[1]) #last part
name_of_file = part_as_string[2]
counter = int(name_of_file[3:name_of_file.find("_")]) - 1
while counter != 0:
index -= 1
part_as_string = messages[index][0].split("#split:#")
if len(part_as_string
) == 4 and part_as_string[0] == sender_of_file:
content_of_file.append(part_as_string[1])
counter -= 1
type_of_file = ""
if name_of_file[0:3] == "pdf":
type_of_file = "pdf"
elif name_of_file[0:3] == "img":
type_of_file = "img"
content_of_file.reverse()
assembled_content = self.assemble_file_parts(content_of_file)
self.open_file(assembled_content,
name_of_file[name_of_file.find("_") + 1:],
type_of_file)
@staticmethod
def generate_ID(): #generates a secure random id
ID = list(''.join(random.SystemRandom().choice(string.ascii_lowercase +
string.ascii_uppercase +
string.digits)
for i in range(6)))
for i in range(len(ID)):
if ID[i] == 'I' or ID[
i] == 'l': #to increase readibility we do not allow I and l since they aren't differentiatable in out chosen Font
ID[i] = str(random.randint(0, 9))
return (''.join(ID))
@staticmethod
def encode_file(file_path):
# Encode File (to String)
with open(file_path, "rb") as _file:
b64encoded_string = base64.b64encode(_file.read())
file_string = b64encoded_string.decode(
"utf-8") # Convert bytes to String
return file_string
@staticmethod
def open_file(content_of_file, name_of_file, type_of_file):
switch[0] = content_of_file
switch[1] = name_of_file
switch[2] = type_of_file
root2 = Toplevel()
app2 = DisplayFile(root2)
