keii.dev
1# Copyright 2021 Nikita Melekhin. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5class Connection:
6 def __init__(self, name, magic, protected=False):
7 self.name = name
8 self.magic = magic
9 self.protected = protected
10 self.messages = {}
11 self.functions = {}
12
13 def add_message(self, msg_name, params):
14 if msg_name in self.messages:
15 op = self.messages[msg_name]
16 if (params != op):
17 print("{0} has 2 different params".format(msg1_name))
18 exit(1)
19 self.messages[msg_name] = params
20
21 def add_function(self, msg1_name, msg2_name=None):
22 if msg1_name in self.functions:
23 print("{0} has 2 functions".format(msg1_name))
24 exit(1)
25 self.functions[msg1_name] = msg2_name