Ver código fonte

Add second argument to surpress broadcast

Foppe Hemminga 6 anos atrás
pai
commit
81629660b5
1 arquivos alterados com 5 adições e 2 exclusões
  1. 5 2
      main.py

+ 5 - 2
main.py

@@ -13,7 +13,9 @@ if __name__ == '__main__':
             row = int(sys.argv[1])
         except ValueError:
             sys.exit(1)
-
+    broadcast = True
+    if len(sys.argv) > 2:
+        broadcast = False
     start = time.time()
 
     db = database.db
@@ -75,7 +77,8 @@ if __name__ == '__main__':
                                                                                                 scratching.race,
                                                                                                 scratching.horse)
                 print(message)
-                view.broadcast(message)
+                if broadcast:
+                    view.broadcast(message)
                 # store new scratching
                 query = "INSERT INTO horses(venue, race_date, race, horse) VALUES(%s, %s, %s, %s)"
                 cursor.execute(query, (scratching.venue, scratching.date,