From 68134451eed0b765fb567284df8d8c1e4e222fe2 Mon Sep 17 00:00:00 2001
From: Justin Lin <wancat@wancat.cc>
Date: Thu, 14 Mar 2024 14:47:09 +0800
Subject: [PATCH] feat: add start script

---
 main.py  | 36 ++++++++++++++++++++++++++++--------
 omx.py   | 16 ----------------
 relay.py |  3 ---
 start.sh |  5 +++++
 4 files changed, 33 insertions(+), 27 deletions(-)
 delete mode 100644 omx.py
 create mode 100755 start.sh

diff --git a/main.py b/main.py
index 30dfcb1..d36a256 100644
--- a/main.py
+++ b/main.py
@@ -1,17 +1,37 @@
 import time
 from player import Player
+from relay import Relay
 
 video_path = "/home/wancat/mind-blowing.mp4"
+PIN = 12
 
-def main():
+ON = 17
+OFF = 6 * 60 + 5
+
+def wait(player, sec):
+    while player.get_time() < sec * 1000:
+        time.sleep(0.1)
+
+
+def main(on, off):
     player = Player(video_path)
-    while True:
-        player.play()
-        while player.get_time() < 10000:
-            time.sleep(0.1)
-        player.reset()
-
+    relay = Relay(PIN)
+    try:
+        while True:
+            player.play()
+            wait(player, on)
+            relay.turn_on()
+            wait(player, off)
+            relay.turn_off()
+            wait(player, player.get_length() / 1000 - 1)
+            player.reset()
+    except Exception as e:
+        print(e)
+    finally:
+        print("exiting..")
+        player.stop()
+        relay.tear_down()
 
 if __name__ == '__main__':
-    main()
+    main(ON, OFF)
 
diff --git a/omx.py b/omx.py
deleted file mode 100644
index 2bf8842..0000000
--- a/omx.py
+++ /dev/null
@@ -1,16 +0,0 @@
-from omxplayer.player import OMXPlayer
-import time
-
-v1 = '/home/pi/Videos/nicechord-original.mp4'
-v2 = '/home/pi/Videos/new_trunk for Alice(290)_1.mp4'
-
-p1 = OMXPlayer(v1, args=['--display=2'], dbus_name='org.mpris.MediaPlayer2.omxplayer1')
-p2 = OMXPlayer(v2, args=['--display=7'], dbus_name='org.mpris.MediaPlayer2.omxplayer2')
-
-try:
-    time.sleep(10)
-except KeyboardInterrupt:
-    print('exit')
-finally:
-    p1.quit()
-    p2.quit()
diff --git a/relay.py b/relay.py
index 4f5f4c0..86c93ea 100644
--- a/relay.py
+++ b/relay.py
@@ -6,8 +6,6 @@ import time
 class Relay:
     def __init__(self, pin):
         self.pin = pin
-
-    def setup(self):
         GPIO.setmode(GPIO.BOARD)
         GPIO.setup(self.pin, GPIO.OUT, initial=GPIO.HIGH)
 
@@ -22,7 +20,6 @@ class Relay:
 
 if __name__ == '__main__':
     relay = Relay(12)
-    relay.setup()
     relay.turn_on()
     time.sleep(5)
     relay.turn_off()
diff --git a/start.sh b/start.sh
new file mode 100755
index 0000000..3b9c1f6
--- /dev/null
+++ b/start.sh
@@ -0,0 +1,5 @@
+#!/bin/bash
+
+cd /home/wancat/relay &&
+source .venv/bin/activate &&
+python main.py