Skip to content

Commit 3895d5d

Browse files
Format files using DocumentFormat
1 parent aa8dfd9 commit 3895d5d

File tree

4 files changed

+83
-83
lines changed

4 files changed

+83
-83
lines changed

docs/make.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ using Documenter, CSVFiles
33
makedocs(
44
modules = [CSVFiles],
55
sitename = "CSVFiles.jl",
6-
analytics="UA-132838790-1",
6+
analytics = "UA-132838790-1",
77
pages = [
88
"Introduction" => "index.md"
99
]

src/CSVFiles.jl

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -50,36 +50,36 @@ end
5050

5151
Base.showable(::MIME"application/vnd.dataresource+json", source::CSVStream) = true
5252

53-
function fileio_load(f::FileIO.File{FileIO.format"CSV"}, deprecated_delim=nothing; delim=deprecated_delim===nothing ? ',' : deprecated_delim, args...)
54-
if deprecated_delim!==nothing
55-
deprecated_delim!=delim && error("deprecated_delim and delim can not both be used at the same time.")
53+
function fileio_load(f::FileIO.File{FileIO.format"CSV"}, deprecated_delim = nothing; delim = deprecated_delim === nothing ? ',' : deprecated_delim, args...)
54+
if deprecated_delim !== nothing
55+
deprecated_delim != delim && error("deprecated_delim and delim can not both be used at the same time.")
5656
Base.depwarn("The positional `delim` keyword in the `load` function is deprecated. Instead use the keyword argument `delim`.", :CSVFiles)
5757
end
5858

5959
return CSVFile(f.filename, delim, args)
6060
end
6161

62-
function fileio_load(f::FileIO.File{FileIO.format"TSV"}, deprecated_delim=nothing; delim=deprecated_delim===nothing ? '\t' : deprecated_delim, args...)
63-
if deprecated_delim!==nothing
64-
deprecated_delim!=delim && error("deprecated_delim and delim can not both be used at the same time.")
62+
function fileio_load(f::FileIO.File{FileIO.format"TSV"}, deprecated_delim = nothing; delim = deprecated_delim === nothing ? '\t' : deprecated_delim, args...)
63+
if deprecated_delim !== nothing
64+
deprecated_delim != delim && error("deprecated_delim and delim can not both be used at the same time.")
6565
Base.depwarn("The positional `delim` keyword in the `load` function is deprecated. Instead use the keyword argument `delim`.", :CSVFiles)
6666
end
6767

6868
return CSVFile(f.filename, delim, args)
6969
end
7070

71-
function fileio_load(s::FileIO.Stream{FileIO.format"CSV"}, deprecated_delim=nothing; delim=deprecated_delim===nothing ? ',' : deprecated_delim, args...)
72-
if deprecated_delim!==nothing
73-
deprecated_delim!=delim && error("deprecated_delim and delim can not both be used at the same time.")
71+
function fileio_load(s::FileIO.Stream{FileIO.format"CSV"}, deprecated_delim = nothing; delim = deprecated_delim === nothing ? ',' : deprecated_delim, args...)
72+
if deprecated_delim !== nothing
73+
deprecated_delim != delim && error("deprecated_delim and delim can not both be used at the same time.")
7474
Base.depwarn("The positional `delim` keyword in the `load` function is deprecated. Instead use the keyword argument `delim`.", :CSVFiles)
7575
end
7676

7777
return CSVStream(s.io, delim, args)
7878
end
7979

80-
function fileio_load(s::FileIO.Stream{FileIO.format"TSV"}, deprecated_delim=nothing; delim=deprecated_delim===nothing ? '\t' : deprecated_delim, args...)
81-
if deprecated_delim!==nothing
82-
deprecated_delim!=delim && error("deprecated_delim and delim can not both be used at the same time.")
80+
function fileio_load(s::FileIO.Stream{FileIO.format"TSV"}, deprecated_delim = nothing; delim = deprecated_delim === nothing ? '\t' : deprecated_delim, args...)
81+
if deprecated_delim !== nothing
82+
deprecated_delim != delim && error("deprecated_delim and delim can not both be used at the same time.")
8383
Base.depwarn("The positional `delim` keyword in the `load` function is deprecated. Instead use the keyword argument `delim`.", :CSVFiles)
8484
end
8585

@@ -98,9 +98,9 @@ function _loaddata(file)
9898
if startswith(file.filename, "https://") || startswith(file.filename, "http://")
9999
response = HTTP.get(file.filename)
100100
data = String(response.body)
101-
return TextParse._csvread(data, file.delim; stringarraytype=Array, file.keywords...)
101+
return TextParse._csvread(data, file.delim; stringarraytype = Array, file.keywords...)
102102
else
103-
return csvread(file.filename, file.delim; stringarraytype=Array, file.keywords...)
103+
return csvread(file.filename, file.delim; stringarraytype = Array, file.keywords...)
104104
end
105105
end
106106

@@ -114,20 +114,20 @@ end
114114

115115
function TableTraits.get_columns_copy_using_missing(file::CSVFile)
116116
columns, colnames = _loaddata(file)
117-
return NamedTuple{(Symbol.(colnames)...,), Tuple{typeof.(columns)...}}((columns...,))
117+
return NamedTuple{(Symbol.(colnames)...,),Tuple{typeof.(columns)...}}((columns...,))
118118
end
119119

120120
function IteratorInterfaceExtensions.getiterator(s::CSVStream)
121-
res = TextParse.csvread(s.io, s.delim; stringarraytype=Array, s.keywords...)
121+
res = TextParse.csvread(s.io, s.delim; stringarraytype = Array, s.keywords...)
122122

123123
it = TableTraitsUtils.create_tableiterator([i for i in res[1]], [Symbol(i) for i in res[2]])
124124

125125
return it
126126
end
127127

128128
function TableTraits.get_columns_copy_using_missing(s::CSVStream)
129-
columns, colnames = TextParse.csvread(s.io, s.delim; stringarraytype=Array, s.keywords...)
130-
return NamedTuple{(Symbol.(colnames)...,), Tuple{typeof.(columns)...}}((columns...,))
129+
columns, colnames = TextParse.csvread(s.io, s.delim; stringarraytype = Array, s.keywords...)
130+
return NamedTuple{(Symbol.(colnames)...,),Tuple{typeof.(columns)...}}((columns...,))
131131
end
132132

133133
function Base.collect(x::CSVFile)

src/csv_writer.jl

Lines changed: 32 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ end
55
function _writevalue(io::IO, value::AbstractString, delim, quotechar, escapechar, nastring)
66
print(io, quotechar)
77
for c in value
8-
if c==quotechar || c==escapechar
8+
if c == quotechar || c == escapechar
99
print(io, escapechar)
1010
end
1111
print(io, c)
@@ -32,8 +32,8 @@ end
3232
push_exprs = Expr(:block)
3333
for i in 1:n
3434
push!(push_exprs.args, :( _writevalue(io, i.$(col_names[i]), delim, quotechar, escapechar, nastring) ))
35-
if i<n
36-
push!(push_exprs.args, :( print(io, delim ) ))
35+
if i < n
36+
push!(push_exprs.args, :( print(io, delim) ))
3737
end
3838
end
3939
push!(push_exprs.args, :( println(io) ))
@@ -45,61 +45,61 @@ end
4545
end
4646
end
4747

48-
function _save(io, data; delim=',', quotechar='"', escapechar='"', nastring="NA", header=true)
48+
function _save(io, data; delim = ',', quotechar = '"', escapechar = '"', nastring = "NA", header = true)
4949
isiterabletable(data) || error("Can't write this data to a CSV file.")
5050

5151
it = getiterator(data)
5252
colnames = collect(eltype(it).parameters[1])
5353

5454
if header
55-
if quotechar===nothing
56-
join(io,[string(colname) for colname in colnames],delim)
55+
if quotechar === nothing
56+
join(io, [string(colname) for colname in colnames], delim)
5757
else
58-
join(io,["$(quotechar)" * replace(string(colname), quotechar => "$(escapechar)$(quotechar)") * "$(quotechar)" for colname in colnames],delim)
58+
join(io, ["$(quotechar)" * replace(string(colname), quotechar => "$(escapechar)$(quotechar)") * "$(quotechar)" for colname in colnames], delim)
5959
end
6060
println(io)
6161
end
6262
_writecsv(io, it, eltype(it), delim, quotechar, escapechar, nastring)
6363
end
6464

65-
function _save(filename::AbstractString, data; delim=',', quotechar='"', escapechar='"', nastring="NA", header=true)
65+
function _save(filename::AbstractString, data; delim = ',', quotechar = '"', escapechar = '"', nastring = "NA", header = true)
6666
isiterabletable(data) || error("Can't write this data to a CSV file.")
6767

6868
ext = last(split(filename, '.'))
6969

7070
if ext == "gz" # Gzipped
7171
open(GzipCompressorStream, filename, "w") do io
72-
_save(io, data, delim=delim, quotechar=quotechar, escapechar=escapechar, nastring=nastring, header=header)
72+
_save(io, data, delim = delim, quotechar = quotechar, escapechar = escapechar, nastring = nastring, header = header)
7373
end
7474
else
7575
open(filename, "w") do io
76-
_save(io, data, delim=delim, quotechar=quotechar, escapechar=escapechar, nastring=nastring, header=header)
76+
_save(io, data, delim = delim, quotechar = quotechar, escapechar = escapechar, nastring = nastring, header = header)
7777
end
7878
end
7979
end
8080

8181

8282

83-
function fileio_save(f::FileIO.File{FileIO.format"CSV"}, data; delim=',', quotechar='"', escapechar='"', nastring="NA", header=true)
84-
return _save(f.filename, data, delim=delim, quotechar=quotechar, escapechar=escapechar, nastring=nastring, header=header)
83+
function fileio_save(f::FileIO.File{FileIO.format"CSV"}, data; delim = ',', quotechar = '"', escapechar = '"', nastring = "NA", header = true)
84+
return _save(f.filename, data, delim = delim, quotechar = quotechar, escapechar = escapechar, nastring = nastring, header = header)
8585
end
8686

87-
function fileio_save(f::FileIO.File{FileIO.format"TSV"}, data; delim='\t', quotechar='"', escapechar='"', nastring="NA", header=true)
88-
return _save(f.filename, data, delim=delim, quotechar=quotechar, escapechar=escapechar, nastring=nastring, header=header)
87+
function fileio_save(f::FileIO.File{FileIO.format"TSV"}, data; delim = '\t', quotechar = '"', escapechar = '"', nastring = "NA", header = true)
88+
return _save(f.filename, data, delim = delim, quotechar = quotechar, escapechar = escapechar, nastring = nastring, header = header)
8989
end
9090

91-
function fileio_save(s::FileIO.Stream{FileIO.format"CSV"}, data; delim=',', quotechar='"', escapechar='"', nastring="NA", header=true)
92-
return _save(s.io, data, delim=delim, quotechar=quotechar, escapechar=escapechar, nastring=nastring, header=header)
91+
function fileio_save(s::FileIO.Stream{FileIO.format"CSV"}, data; delim = ',', quotechar = '"', escapechar = '"', nastring = "NA", header = true)
92+
return _save(s.io, data, delim = delim, quotechar = quotechar, escapechar = escapechar, nastring = nastring, header = header)
9393
end
9494

95-
function fileio_save(s::FileIO.Stream{FileIO.format"TSV"}, data; delim='\t', quotechar='"', escapechar='"', nastring="NA", header=true)
96-
return _save(s.io, data, delim=delim, quotechar=quotechar, escapechar=escapechar, nastring=nastring, header=header)
95+
function fileio_save(s::FileIO.Stream{FileIO.format"TSV"}, data; delim = '\t', quotechar = '"', escapechar = '"', nastring = "NA", header = true)
96+
return _save(s.io, data, delim = delim, quotechar = quotechar, escapechar = escapechar, nastring = nastring, header = header)
9797
end
9898

9999
#
100100
# Streaming version writes header (if any) on first call, then appends on subsequent calls.
101101
#
102-
const CSV_or_TSV = Union{FileIO.format"CSV", FileIO.format"TSV"}
102+
const CSV_or_TSV = Union{FileIO.format"CSV",FileIO.format"TSV"}
103103

104104
_delim(T) = T <: FileIO.format"CSV" ? ',' : '\t'
105105

@@ -112,30 +112,30 @@ mutable struct CSVFileSaveStream{T}
112112
nastring::AbstractString
113113
header::Bool
114114
end
115-
116-
function fileio_savestreaming(f::FileIO.File{T}, data=nothing; delim=_delim(T), quotechar='"', escapechar='"', nastring="NA",
117-
header=true) where T <: CSV_or_TSV
115+
116+
function fileio_savestreaming(f::FileIO.File{T}, data = nothing; delim = _delim(T), quotechar = '"', escapechar = '"', nastring = "NA",
117+
header = true) where T <: CSV_or_TSV
118118
io = open(f.filename, "w")
119119

120-
if data!==nothing
121-
_save(io, data; delim=delim, quotechar=quotechar, escapechar=escapechar, nastring=nastring, header=header)
120+
if data !== nothing
121+
_save(io, data; delim = delim, quotechar = quotechar, escapechar = escapechar, nastring = nastring, header = header)
122122
end
123123

124-
return CSVFileSaveStream(io, data!==nothing, delim, quotechar, escapechar, nastring, header)
124+
return CSVFileSaveStream(io, data !== nothing, delim, quotechar, escapechar, nastring, header)
125125
end
126126

127-
function fileio_savestreaming(s::FileIO.Stream{T}, data=nothing; delim=_delim(T), quotechar='"', escapechar='"', nastring="NA",
128-
header=false) where T <: CSV_or_TSV
127+
function fileio_savestreaming(s::FileIO.Stream{T}, data = nothing; delim = _delim(T), quotechar = '"', escapechar = '"', nastring = "NA",
128+
header = false) where T <: CSV_or_TSV
129129

130-
if data!==nothing
131-
_save(s.io, data; delim=delim, quotechar=quotechar, escapechar=escapechar, nastring=nastring, header=header)
130+
if data !== nothing
131+
_save(s.io, data; delim = delim, quotechar = quotechar, escapechar = escapechar, nastring = nastring, header = header)
132132
end
133-
134-
return CSVFileSaveStream(s.io, data!==nothing, delim, quotechar, escapechar, nastring, header)
133+
134+
return CSVFileSaveStream(s.io, data !== nothing, delim, quotechar, escapechar, nastring, header)
135135
end
136136

137137
function Base.write(s::CSVFileSaveStream, data)
138-
_save(s.io, data; delim=s.delim, quotechar=s.quotechar, escapechar=s.escapechar, nastring=s.nastring, header=s.first_data_written ? false : header)
138+
_save(s.io, data; delim = s.delim, quotechar = s.quotechar, escapechar = s.escapechar, nastring = s.nastring, header = s.first_data_written ? false : header)
139139
end
140140

141141
function Base.close(s::CSVFileSaveStream)

0 commit comments

Comments
 (0)