projects
/
quanlib.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Stores book metadata to PostgreSQL database.
[quanlib.git]
/
main.rb
diff --git
a/main.rb
b/main.rb
index 2d7f68b25c33c6bf45dbb0e909e17e3146f53e5b..bc175fbf51ae33205a37735d1390872a3ea630e1 100644
(file)
--- a/
main.rb
+++ b/
main.rb
@@
-1,3
+1,4
@@
+require 'store'
require 'walkdir'
outputDir = 'output'
require 'walkdir'
outputDir = 'output'
@@
-5,11
+6,34
@@
outputDir = 'output'
books = []
imageCount = 0
books = []
imageCount = 0
+def handleArg(arg)
+ if "--purge" == arg
+ puts 'Purging database...'
+ @store.dropSchema()
+ elsif arg.start_with?("--")
+ abort('ERROR: Unrecognized option "' + arg + '".')
+ end
+end
+
+@store = Store.new()
+@store.connect()
+
+for arg in ARGV
+ handleArg(arg)
+end
+
+@store.init_db()
+
for arg in ARGV
for arg in ARGV
- w = WalkDir.new(arg)
- books += (w.books)
+ if ! arg.start_with?("--")
+ puts 'Scanning directory "' + arg + '"...'
+ w = WalkDir.new(@store, arg)
+ books += (w.books)
+ end
end
end
+puts 'Creating output...'
+
if ! Dir.exist?(outputDir)
Dir.mkdir(outputDir)
end
if ! Dir.exist?(outputDir)
Dir.mkdir(outputDir)
end
@@
-65,3
+89,4
@@
open(outputDir + '/index.html', 'w') do |fd|
fd.puts "</html>"
end
fd.puts "</html>"
end
+@store.disconnect()