在Ruby中查找文件名的扩展名

Question

我正在处理Rails应用程序的文件上传部分.应用程序以不同方式处理不同类型的文件.

我想制作某些文件扩展名的白名单来检查上传的文件,看看它们应该去哪里.所有文件名都是字符串.

我需要一种方法来只检查文件名字符串的扩展部分.文件名的格式均为"some_file_name.some_extension".

Answer 1

这是非常基本的东西:

irb(main):002:0> accepted_formats = [".txt", ".pdf"]
=> [".txt", ".pdf"]
irb(main):003:0> File.extname("example.pdf") # get the extension
=> ".pdf"
irb(main):004:0> accepted_formats.include? File.extname("example.pdf")
=> true
irb(main):005:0> accepted_formats.include? File.extname("example.txt")
=> true
irb(main):006:0> accepted_formats.include? File.extname("example.png")
=> false

Answer 2

使用extnameFile类中的方法

File.extname("test.rb")         #=> ".rb"

你也可能需要basename方法

File.basename("/home/gumby/work/ruby.rb", ".rb")   #=> "ruby"

Answer 3

相当古老的主题,但这里是摆脱扩展分隔符点和可能的尾随空格的方法:

File.extname(path).strip.downcase[1..-1]

例子:

File.extname(".test").strip.downcase[1..-1]       # => nil
File.extname(".test.").strip.downcase[1..-1]      # => nil
File.extname(".test.pdf").strip.downcase[1..-1]   # => "pdf"
File.extname(".test.pdf ").strip.downcase[1..-1]  # => "pdf"

Answer 4

我认为这样做会更容易摆脱扩展分隔符。

File.extname(path).delete('.')