improvements to code style in gen_index

This commit is contained in:
Christoffer Müller Madsen 2017-05-09 12:02:15 +02:00
parent 0c9405b714
commit ab55e98974

View File

@ -1,39 +1,33 @@
require 'erb' require 'erb'
require 'filesize' require 'filesize'
RES_BASE = "https://strawberry.thedevcave.net/~christoffermadsen/"
usernames = ["christoffermadsen","jmaa","jacob","alexander"]
shot_dir = "/var/shots/"
erb_file = "dcav-index.erb" erb_file = "dcav-index.erb"
html_file = "/var/shots/html/index.html" html_file = "/var/shots/html/index.html"
class User class User
attr_accessor :name, :count, :size attr_accessor :name, :count, :size
def <=> (other) def <=> other
other.count <=> count other.count <=> count
end end
end end
RES_BASE = "https://strawberry.thedevcave.net/~christoffermadsen/"
usernames = ["christoffermadsen","jmaa","jacob","alexander"]
users = [] users = []
shot_dir = "/var/shots/"
usernames.each do |username| usernames.each do |username|
user = User.new user = User.new
user.name = username user.name = username
size = `du -B1 #{shot_dir}#{username}/ | awk {'print $1'} | head -n 1` user.size = `du -B1 #{shot_dir}#{username}/ | awk {'print $1'} | head -n 1`
count = `ls -1 #{shot_dir}#{username}/ | wc -l | head -n 1` .delete("\n")
.to_i
user.count = `ls -1 #{shot_dir}#{username}/ | wc -l | head -n 1`
.delete("\n")
.to_i
[size,count].each do |x|
x.delete! "\n"
end
user.size = size.to_i
user.count = count.to_i
users << user users << user
end end
output = ERB.new(File.read(erb_file)).result
File.open(html_file, 'w') do |f| File.open(html_file, 'w') do |f|
f.write(output) f << ERB.new(File.read erb_file).result
end end