summaryrefslogtreecommitdiff
path: root/lib/pytst.c
blob: a760318d2d82357aa36132a2520c70c02a0c5f1b (plain)
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

#include "python.h"


#include "config.h"
#include "common.h"
#include "db.h"
#include "epgservice.h"


cDbTable* eventsDb = 0;
cDbConnection* connection = 0;
const char* logPrefix = "";

//***************************************************************************
// Init / Exit
//***************************************************************************

void initConnection()
{
   cDbConnection::init();

   cDbConnection::setEncoding("utf8");
   cDbConnection::setHost("localhost");

   cDbConnection::setPort(3306);
   cDbConnection::setName("epg2vdr");
   cDbConnection::setUser("epg2vdr");
   cDbConnection::setPass("epg");
   cDbConnection::setConfPath("/etc/epgd/");

   connection = new cDbConnection();
}

void exitConnection()
{
   cDbConnection::exit();

   if (connection)
      delete connection;
}

int init()
{
   eventsDb = new cDbTable(connection, "useevents");
   if (eventsDb->open() != success) return fail;

   return success;
}

int exit()
{
   delete eventsDb;
   return done;
}

//***************************************************************************
// Main
//***************************************************************************

int main(int argc, char** argv)
{
   cEpgConfig::logstdout = yes;
   cEpgConfig::loglevel = 0;
   int namingmode = tnmAuto;
   const char* tmplExpression = "";

   if (argc < 4)
   {
      tell(0, "Usage: pytst <cnt_source> <cnt_channnelid> <cn_eventid> [<namingmode>] [<template>]");
      tell(0, " Get the data e.g. with:");
      tell(0, "   %s", "mysql -u epg2vdr -pepg -Depg2vdr --default-character-set=utf8' epgdb -e \"select cnt_source, cnt_eventid, cnt_channelid, sub_title, sub_shorttext from useevents where sub_title like '%whatever%';\"");
      return 1;
   }

   if (argc >= 5)
      namingmode = atoi(argv[4]);

   if (argc >= 6)
      tmplExpression = argv[5];

   // at first allpy locale !!

   setlocale(LC_CTYPE, "");

   // read dictionary

   if (dbDict.in("/etc/epgd/epg.dat") != success)
   {
      tell(0, "Invalid dictionary configuration, aborting!");
      return 1;
   }

   initConnection();
   init();

   eventsDb->clear();
   eventsDb->setValue("CNTSOURCE", argv[1]);
   eventsDb->setValue("CHANNELID", argv[2]);
   eventsDb->setBigintValue("CNTEVENTID", atol(argv[3]));

   if (!eventsDb->find())
   {
      tell(0, "Event %s/%ld not found", argv[2], atol(argv[3]));
      return 1;
   }

   tell(2, "Event '%s/%s' found",
        eventsDb->getStrValue("TITLE"), eventsDb->getStrValue("SHORTTEXT"));

   // Python stuff ..

   Python py("recording", "name");

   if (py.init("/etc/epgd") != success)
   {
      tell(0, "Init of python failed!");
      return 1;
   }

   if (py.execute(eventsDb, namingmode, tmplExpression) == success)
      tell(0, "Info: The recording name calculated by 'recording.py' (with namingmode %d) is '%s'", namingmode, py.getResult());

   py.exit();
   exitConnection();

   return 0;
}