Как читать и записывать регистры из файла CSV (в C)? - PullRequest
Новая
       17

Как читать и записывать регистры из файла CSV (в C)?

1 голос
/ 02 апреля 2019

У меня есть файл .CSV, и мне нужно прочитать его и записать в двоичный файл.

Я пытался адаптировать некоторые коды, которые я видел в некоторых похожих вопросах, но не сработал.

Мой файл похож на: 

nroInscricao,nota,data,cidade,nomeEscola
439,607.5,01/01/2004,Maceio,PEDRO II
387,,,Sao Paulo,JOAO KOPKE
332,400.8,03/01/2004,Brasilia,REINALDO RIBEIRO DA SILVA DOU
296,436.4,04/01/2004,,JOSE CANDIDO DE SOUZA

И япопытался прочитать файл со следующим кодом: 

const char* getfield(char* line, int num){

    const char* tok;
    for (tok = strtok(line, ","); tok && *tok; tok = strtok(NULL, ",\n")){
        if (!--num)
            return tok;
    }
    return NULL;

}


int main(){

    FILE* stream = fopen("C:\\Users\\10734140\\Downloads\\SCC0503012019trabalho1.csv", "r+");

    char line[1024];
    while (fgets(line, 1024, stream)){
        char* tmp = strdup(line);
        printf("Field 3 would be %s\n", getfield(tmp, 3));
        // NOTE strtok clobbers tmp
        free(tmp);
    }
}

Но ничего не получилось в исполнении.

1 Ответ

2 голосов
/ 02 апреля 2019

Просто напишите свой strtokEvenEmpty , полученный из strtok , затем замените 

     for (tok = strtok(line, ","); tok && *tok; tok = strtok(NULL, ",\n")){

на 

for (tok = strtokEvenEmpty(line, ","); tok != NULL; tok = strtokEvenEmpty(NULL, ",\n")){

, потому чтоваш тест *tok останавливает вас, когда поле пусто даже до ожидаемого поля

Например: 

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

char * strtokEvenEmpty(char * s, const char * seps)
{
  static char * p = NULL;

  if (s != NULL)
    p = s;
  else if (p == NULL)
    return NULL;
  else
    s = p;

  while (*p) {
    if (strchr(seps, *p)) {
      *p++ = 0;
      return s;
    }
    p += 1;
  }
  return (*s) ? s : NULL;
}

const char * getfield(char* line, int num){
  const char * tok;

  for (tok = strtokEvenEmpty(line, ","); tok; tok = strtokEvenEmpty(NULL, ",\n")){
    if (!--num)
      return tok;
  }
  return NULL;
}

int main()
{
  FILE * stream = fopen("SCC0503012019trabalho1.csv", "r");

  if (stream != NULL) {
    char line[1024];
    while (fgets(line, 1024, stream)) {
      printf("Field 3 would be '%s'\n", getfield(line, 3));
    }
    fclose(stream);
  }
}

Компиляция и выполнение: 

pi@raspberrypi:/tmp $ gcc -pedantic -Wextra -Wall -g s.c
pi@raspberrypi:/tmp $ cat SCC0503012019trabalho1.csv 
nroInscricao,nota,data,cidade,nomeEscola
439,607.5,01/01/2004,Maceio,PEDRO II
387,,,Sao Paulo,JOAO KOPKE
332,400.8,03/01/2004,Brasilia,REINALDO RIBEIRO DA SILVA DOU
296,436.4,04/01/2004,,JOSE CANDIDO DE SOUZA
pi@raspberrypi:/tmp $ ./a.out
Field 3 would be 'data'
Field 3 would be '01/01/2004'
Field 3 would be ''
Field 3 would be '03/01/2004'
Field 3 would be '04/01/2004'

И еслиЯ получаю четвертое поле, а не третье (printf("Field 4 would be %s\n", getfield(line, 4));): 

Field 4 would be 'cidade'
Field 4 would be 'Maceio'
Field 4 would be 'Sao Paulo'
Field 4 would be 'Brasilia'
Field 4 would be ''

В данном случае ваш strdup в main бесполезен, потому чтофакт line изменен с помощью strtok / strtokEvenEmpty не является проблемой, а также факт line , поэтому результат getfield изменяется с помощьюследующий цикл.

...