1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
#!/usr/bin/env python
"""
usage: ./convert_papermode_to_metadata.py
This script converts the .paperinfo CSV file in the current directory to an
equivalent .metadata.json file.
ranger used to store metadata in .paperinfo files, but that format was rather
limited, so .metadata.json files were introduced.
"""
from __future__ import (absolute_import, division, print_function)
import csv
import json
import os
import sys
if sys.version_info[0] < 3:
input = raw_input # NOQA pylint: disable=undefined-variable,redefined-builtin,invalid-name
FIELDS = ["name", "year", "title", "authors", "url"]
def replace(source, target):
if not os.path.exists(source):
print("Source file `%s' doesn't exist, skipping." % source)
return
# Ask for user confirmation if the target file already exists
if os.path.exists(target):
sys.stdout.write("Warning: target file `%s' exists! Overwrite? [y/N]")
userinput = input()
if not (userinput.startswith("y") or userinput.startswith("Y")):
print("Skipping file `%s'" % source)
return
result = {}
# Read the input file and convert it to a dictionary
with open(".paperinfo", "r") as infile:
reader = csv.reader(infile, skipinitialspace=True)
for lineno, row in enumerate(reader):
if len(row) != len(FIELDS):
print("skipping invalid row `%s' on line %d" % (row, lineno))
continue
name = row[0]
entry = {}
# Filling up the resulting entry dict
for i, column in enumerate(row[1:]):
if column:
entry[FIELDS[i + 1]] = column
# Adding the dict if it isn't empty
if entry:
result[name] = entry
# Write the obtained dictionary into the target file
if result:
# There's no way to specify encoding in 2.x even though in this case we
# could choose to write in UTF-8.
with open(".metadata.json", "w") as outfile:
json.dump(result, outfile, indent=2)
else:
print("Skipping writing `%s' due to a lack of data" % target)
if __name__ == "__main__":
if set(['--help', '-h']) & set(sys.argv[1:]):
print(__doc__.strip())
else:
replace(".paperinfo", ".metadata.json")
|
