Check for road wins before flat wins

1 files changed, 7 insertions, 8 deletions

diff --git a/extract.sh b/extract.sh
index e091a9c..d351244 100755
--- a/extract.sh
+++ b/extract.sh
@@ -4,20 +4,19 @@ db_file=games_anon.db
 
 query() {
 		echo "SELECT $1 FROM games WHERE (size == $2) \
-and (result != '1-0') and (result != '0-1') and (result != '0-0')\
-and (result != '1/2-1/2');"
+and ((result == 'R-0') or (result == '0-R'));"
+# and (result != '1-0') and (result != '0-1') and (result != '0-0')\
+# and (result != '1/2-1/2');"
 }
 
 extract() {
-		cd data
-		if [ ! -f "$db_file" ]; then
-				wget "https://www.playtak.com/games_anon.db"
+		if [ ! -f "data/$db_file" ]; then
+				wget "https://www.playtak.com/games_anon.db" -O "data/$db_file"
 		fi
-		for size in 5 6; do
+		for size in 5; do
 				echo Extracing games of size "$size"...
-				sqlite3 "$db_file" "$(query $1 $size)" | shuf > "playtak-$size"
+				sqlite3 "data/$db_file" "$(query $1 $size)" | shuf > "data/playtak-$size"
 		done
-		cd ..
 }
 
 process() {