Merge pull request #44 from SamuelGuillemet/feature/gui_charts

Minor improvement for GUI

clients/graphical-user-interface/quickfix_client_gui/db/database_manager.py

@@ -100,6 +100,13 @@ def init_example_data(self):
             self.session.add_all([order1, order2])
             self.session.commit()
 
+    def create_user(self, username):
+        user = self.session.query(User).filter_by(username=username).first()
+        if user is None:
+            user1 = User(username=username, balance=1000)
+            self.session.add(user1)
+            self.session.commit()
+
     def get_user_balance(self, username):
         user = self.session.query(User).filter_by(username=username).first()
 

clients/graphical-user-interface/quickfix_client_gui/interface/main_interface.py

@@ -47,6 +47,7 @@ def __init__(self, username, application, initiator):
         # Initialize class attributes
         self.database_manager = DatabaseManager("sqlite:///quickfix_client_database.db")
         self.username = username
+        self.database_manager.create_user(username)
         self.account_balance = self.database_manager.get_user_balance(self.username)
         self.owned_shares = self.database_manager.get_user_shares(self.username)
         self.application, self.initiator = application, initiator
@@ -208,7 +209,7 @@ def refresh_main_interface(self):
 
         # Refresh the chart
         symbol = self.symbol_entry.get()
-        if len(symbol) in range(1, 5):
+        if len(symbol) in range(1, 6):
             timestamp = int(datetime.datetime.now().strftime("%Y%m%d%H%M"))
             market_data_request_snapshot = MarketDataRequest.new_snapshot_request(
                 timestamp,