Last active
November 3, 2022 14:55
-
-
Save leonardb/dcb1ac05492568afd6e69eb22c2db2d9 to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| -module(fmt_csv). | |
| %% API | |
| -export([row/1]). | |
| row(Row) when is_tuple(Row) -> | |
| row(tuple_to_list(Row)); | |
| row(Row) -> | |
| [string:join([case Field of | |
| null -> ""; | |
| true -> "1"; | |
| false -> "0"; | |
| {A,B,C,D} = IpAddress | |
| when is_integer(A), | |
| is_integer(B), | |
| is_integer(C), | |
| is_integer(D) -> | |
| inet_parse:ntoa(IpAddress); | |
| {A,B,C,D,E,F,G,H} = IpAddress | |
| when is_integer(A), is_integer(B), | |
| is_integer(C), is_integer(D), | |
| is_integer(E), is_integer(F), | |
| is_integer(G), is_integer(H) -> | |
| inet_parse:ntoa(IpAddress); | |
| {{_,_,_},{_,_,_}} = DateTime -> | |
| format_datetime(DateTime); | |
| {Long, Lat} when is_number(Long), is_number(Lat) -> | |
| "(" ++ to_list(Long) ++ ":" ++ to_list(Lat) ++ ")"; | |
| {_, _, _} = Date -> | |
| format_date(Date); | |
| _ when is_number(Field); | |
| is_list(Field); | |
| is_atom(Field); | |
| is_binary(Field) -> | |
| SField = to_list(Field), | |
| StrField = string:strip(SField), | |
| case lists:any(fun($,) -> true; | |
| ($\n) -> true; | |
| ($\r) -> true; | |
| ($") -> true; | |
| (_) -> false | |
| end, StrField) of | |
| true -> | |
| [$", csv_escape_field(StrField), $"]; | |
| false -> | |
| StrField | |
| end; | |
| _Other -> | |
| io:format("unhandled Field value discarded: ~p~n", [Field]), | |
| "" | |
| end || Field <- Row], ","), $\n]. | |
| csv_escape_field(Field) -> | |
| [case Ch of | |
| $" -> [$", $"]; | |
| $\n -> $ ; | |
| _ -> Ch | |
| end || Ch <- Field]. | |
| format_datetime(Time) when is_binary(Time) -> | |
| Time; | |
| format_datetime(Time) -> | |
| {{YYYY, MM, DD}, {H, M, S}} = | |
| case Time of | |
| {_,_,_} -> calendar:now_to_universal_time(Time); | |
| {{_,_,_},{_,_,_}} -> Time | |
| end, | |
| io_lib:format("~4..0b-~2..0b-~2..0b ~2..0b:~2..0b:~2..0b", | |
| [YYYY, MM, DD, H, M, round(S)]). | |
| format_date({Y, M, D}) -> | |
| io_lib:format("~4..0b-~2..0b-~2..0b", [Y, M, D]); | |
| format_date({{Y, M, D}, {_, _, _}}) -> | |
| io_lib:format("~4..0b-~2..0b-~2..0b", [Y, M, D]). | |
| to_list(V) -> | |
| case V of | |
| V when is_list(V) -> | |
| V; | |
| V when is_float(V) -> | |
| float_to_list(V); | |
| V when is_integer(V) -> | |
| integer_to_list(V); | |
| V when is_atom(V) -> | |
| atom_to_list(V); | |
| V when is_binary(V) -> | |
| binary_to_list(V); | |
| V when is_tuple(V) -> | |
| tuple_to_list(V); | |
| V when is_map(V) -> | |
| maps:to_list(V) | |
| end. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment